We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 75b07f4 commit f0e6035Copy full SHA for f0e6035
1 file changed
src/training/trainer.py
@@ -29,7 +29,7 @@ def train():
29
learning_rate=settings.LEARNING_RATE,
30
logging_steps=settings.LOGGING_STEPS,
31
num_train_epochs=settings.NUM_TRAIN_EPOCHS,
32
- eval_strategy="steps",
+ evaluation_strategy="steps",
33
eval_steps=settings.EVAL_STEPS,
34
save_strategy="steps",
35
save_steps=settings.SAVE_STEPS,
0 commit comments