Fix compatibility issue

camlab-ethz · Jun 18, 2024 · d3fcf01 · d3fcf01
1 parent 89e768b
commit d3fcf01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scOT/train.py b/scOT/train.py
@@ -277,7 +277,7 @@ def clean_yaml(config):
     train_config = TrainingArguments(
         output_dir=ckpt_dir,
         overwrite_output_dir=True,  #! OVERWRITE THIS DIRECTORY IN CASE, also for resuming training
-        eval_strategy="epoch",
+        evaluation_strategy="epoch",
         per_device_train_batch_size=config["batch_size"],
         per_device_eval_batch_size=config["batch_size"],
         eval_accumulation_steps=16,