use eval_strategy instead to avoid transformer warning

ibm-granite · Dec 5, 2024 · 5719f13 · 5719f13
1 parent 46d3998
commit 5719f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/services/finetuning/tsfmfinetuning/finetuning.py b/services/finetuning/tsfmfinetuning/finetuning.py
@@ -203,7 +203,7 @@ def _finetuning_common(
             overwrite_output_dir=True,
             learning_rate=parameters.trainer_args.learning_rate,
             num_train_epochs=parameters.trainer_args.num_train_epochs,
-            evaluation_strategy="epoch",
+            eval_strategy="epoch",
             per_device_train_batch_size=parameters.trainer_args.per_device_train_batch_size,
             per_device_eval_batch_size=parameters.trainer_args.per_device_eval_batch_size,
             dataloader_num_workers=4,