Skip to content

Commit

Permalink
update filename for mnist fsdp test
Browse files Browse the repository at this point in the history
Signed-off-by: ashors1 <[email protected]>
  • Loading branch information
ashors1 committed Nov 22, 2024
1 parent ea5ae7a commit 050de36
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/collections/llm/test_mnist_model_nemo2_fsdp.py
Original file line number Diff line number Diff line change
Expand Up @@ -525,6 +525,7 @@ def run_train_mnist_litautoencoder_with_fsdp_strategy_single_gpu():
every_n_train_steps=5,
# Enables the .nemo file-like checkpointing where all IOMixins are under SerDe
always_save_context=True,
filename="{model_name}--{val_loss:.2f}-{step}-{consumed_samples}",
)
root_dir = tmpdir
save_dir = root_dir / name
Expand Down

0 comments on commit 050de36

Please sign in to comment.