Skip to content

Commit

Permalink
mention evaluation steps in TSS docs (#657)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #657

Mention that training **and** evaluation steps are combined when forming the checkpoint name

Reviewed By: galrotem

Differential Revision: D52224894

fbshipit-source-id: 4545486bc8e19a3a3336752f053db2c3fa474a4e
  • Loading branch information
JKSenthil authored and facebook-github-bot committed Dec 18, 2023
1 parent a838566 commit 22abb8c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchtnt/framework/callbacks/torchsnapshot_saver.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ class TorchSnapshotSaver(BaseCheckpointer):
If used with :func:`torchtnt.framework.fit`, this class will also save the evaluation progress state.
Checkpoints will be saved under ``dirpath/epoch_{epoch}_step_{step}`` where step is the *total* number of training steps completed across all epochs.
Checkpoints will be saved under ``dirpath/epoch_{epoch}_step_{step}`` where step is the *total* number of training and evaluation steps completed across all epochs.
Args:
dirpath: Parent directory to save snapshots to.
Expand Down

0 comments on commit 22abb8c

Please sign in to comment.