Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_l
|
|
56 |
# Define the training arguments
|
57 |
training_args = TrainingArguments(
|
58 |
output_dir='./results',
|
59 |
-
|
60 |
learning_rate=2e-5,
|
61 |
per_device_train_batch_size=8,
|
62 |
per_device_eval_batch_size=8,
|
|
|
56 |
# Define the training arguments
|
57 |
training_args = TrainingArguments(
|
58 |
output_dir='./results',
|
59 |
+
eval_strategy='epoch', # Evaluate after each epoch
|
60 |
learning_rate=2e-5,
|
61 |
per_device_train_batch_size=8,
|
62 |
per_device_eval_batch_size=8,
|