Skip to content

Commit

Permalink
Fix failing test by setting the right ckpt filename
Browse files Browse the repository at this point in the history
Signed-off-by: Shriya Palsamudram <spalsamudram@nvidia.com>
  • Loading branch information
ShriyaPalsamudram committed Sep 27, 2024
1 parent d100b16 commit f27c786
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/collections/llm/test_mnist_model_nemo2.py
Original file line number Diff line number Diff line change
Expand Up @@ -501,6 +501,7 @@ def run_train_mnist_litautoencoder_with_megatron_strategy_single_gpu():
monitor="val_loss",
save_top_k=1,
every_n_train_steps=5,
filename="{model_name}--{val_loss:.2f}-{step}-{consumed_samples}",
# Enables the .nemo file-like checkpointing where all IOMixins are under SerDe
always_save_context=True,
)
Expand Down

0 comments on commit f27c786

Please sign in to comment.