flax
f92f936b - Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps.

Commit

4 years ago

Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps. PiperOrigin-RevId: 395424856

References

#1523 - Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps.

Author

tutmann

Committer

a-googler

Parents

4d7d4580

flax f92f936b - Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps.

flax
f92f936b - Add optional keep_every_n_steps argument to checkpointing. When enabled, save_checkpoint will keep checkpoints every n steps.