fix bug in pydantic

NVIDIA · Feb 6, 2025 · ecc9f67 · ecc9f67
1 parent c204616
commit ecc9f67
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/sub-packages/bionemo-llm/src/bionemo/llm/train.py b/sub-packages/bionemo-llm/src/bionemo/llm/train.py
@@ -252,8 +252,6 @@ def train(
         config=bionemo_model_config,
         tokenizer=data.tokenizer,
         optimizer=optimizer,
-        train_metric=None,
-        valid_metric=None,
     )
     trainer: nl.Trainer = setup_trainer(parallel_config, training_config, nsys_config=nsys_config)
     nemo_logger: nl.NeMoLogger = nemo_logger_factory(experiment_config, wandb_config=wandb_config)