Revert "increase limit_val_batches and val_check_interval to avoid du…

…plicated checkpoint_dir name" This reverts commit b936fda.
NVIDIA · Nov 5, 2024 · 5e82404 · 5e82404
1 parent 1d66fbb
commit 5e82404
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/protein/esm2/test_esm2_pretrain.py b/scripts/protein/esm2/test_esm2_pretrain.py
@@ -106,8 +106,8 @@ def test_main_runs(monkeypatch, tmpdir, dummy_protein_dataset, dummy_parquet_tra
             wandb_offline=True,
             num_steps=55,
             warmup_steps=5,
-            limit_val_batches=5,
-            val_check_interval=10,
+            limit_val_batches=1,
+            val_check_interval=1,
             log_every_n_steps=None,
             num_dataset_workers=1,
             biobert_spec_option=BiobertSpecOption.esm2_bert_layer_with_transformer_engine_spec,