Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -105,8 +105,8 @@ if (runModel=='1'):
|
|
105 |
training_args = TrainingArguments(
|
106 |
output_dir='./results_' + modelNameToUse,
|
107 |
num_train_epochs=5,
|
108 |
-
per_device_train_batch_size=
|
109 |
-
per_device_eval_batch_size=
|
110 |
warmup_steps=500,
|
111 |
weight_decay=0.02,
|
112 |
logging_dir='./logs_' + modelNameToUse,
|
|
|
105 |
training_args = TrainingArguments(
|
106 |
output_dir='./results_' + modelNameToUse,
|
107 |
num_train_epochs=5,
|
108 |
+
per_device_train_batch_size=8,
|
109 |
+
per_device_eval_batch_size=8,
|
110 |
warmup_steps=500,
|
111 |
weight_decay=0.02,
|
112 |
logging_dir='./logs_' + modelNameToUse,
|