How to change the frequency of checkpoint saving in Tensorflow API

39 views Asked by At

I'm training a model using Tensorflow API. It saves a checkpoint every 1000 steps. However, I would like to reduce that to 500 steps, and I can't find the place to configure it.

I would appreciate some help to find a way to change configuration settings.

1

There are 1 answers

0
ela16 On BEST ANSWER

It's very simple. Just add this flag to your training script:

--checkpoint_every_n=500

For example:

python {training_script} --model_dir={checkpoint_path} --pipeline_config_path={pipeline_config_path} --num_train_steps=20000 --checkpoint_every_n=500