diff --git a/tests/collections/llm/test_mnist_model_nemo2_fsdp.py b/tests/collections/llm/test_mnist_model_nemo2_fsdp.py index da45d0e1fc38..430613e571f7 100644 --- a/tests/collections/llm/test_mnist_model_nemo2_fsdp.py +++ b/tests/collections/llm/test_mnist_model_nemo2_fsdp.py @@ -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