Give an option for the Checkpoint frequency

I want to suggest an option for the checkpoint frequency, i.e. we can change the checkpoints to save at 1000 batches or 100000 batches or any number we want. Where should the option be present in train_elmo.py?
if batch_no % Batches_After_Checkpoint== 0:
# write the summaries to tensorboard and display perplexity
summary_writer.add_summary(ret[1], batch_no)
print("Batch %s, train_perplexity=%s" % (batch_no, ret[2]))
print("Total time: %s" % (time.time() - t1))