OpenNMT Forum

Saving checkpoint at specified steps

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