Hi,
I am running my training in opennmt-tf for 1,000,000 steps, and I want to save the ckpts at specified steps (say 300k , 500k , 1000k).
Can you please suggest , how I can save automatically the ckpts at this particular steps?
Currently I am doing it manually , which I want to avoid for the purpose of automation.
Thanks,
Sr
Dear Sriram,
Saving checkpoints at specific steps is not possible per ce; however, there are similar solutions. The options that are related to checkpoints in OpenNMT-tf are as follows:
Save a checkpoint every this many steps (default: 5000).
save_checkpoints_steps: null
How many checkpoints to keep on disk.
keep_checkpoint_max: 3
Maybe the first option can server your needs.
Kind regards,
Yasmin