piska0

app.py

@@ -34,12 +34,13 @@ model = AutoModelForSequenceClassification.from_pretrained(
 
 # Задаем параметры обучения
 training_args = TrainingArguments(
-	output_dir = 'test_trainer_log',
-	evaluation_strategy = 'epoch',
-	per_device_train_batch_size = 6,
-	per_device_eval_batch_size = 6,
-	num_train_epochs = 5,
-	report_to='none')
+    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,
+    report_to='none'
+)
 
 # Определяем как считать метрику
 metric = evaluate.load('f1')