Spaces:

SidorCrew
/

GigachatProj

Runtime error

AritORR commited on Jul 10

Commit

a165845

1 Parent(s): 3bbb15e

piska1

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ model = AutoModelForSequenceClassification.from_pretrained(
 # Задаем параметры обучения
 training_args = TrainingArguments(
     output_dir='test_trainer_log',
-    evaluate_during_training=True,
     per_device_train_batch_size=6,
     per_device_eval_batch_size=6,
     num_train_epochs=5,

 # Задаем параметры обучения
 training_args = TrainingArguments(
     output_dir='test_trainer_log',
+    eval_strategy='epoch',
     per_device_train_batch_size=6,
     per_device_eval_batch_size=6,
     num_train_epochs=5,