Update app.py
Browse files
app.py
CHANGED
|
@@ -214,9 +214,9 @@ training_args = Seq2SeqTrainingArguments(
|
|
| 214 |
evaluation_strategy='epoch',
|
| 215 |
per_device_train_batch_size=TrainingConfig.BATCH_SIZE,
|
| 216 |
per_device_eval_batch_size=TrainingConfig.BATCH_SIZE,
|
| 217 |
-
fp16=
|
| 218 |
-
fp16_full_eval=True,
|
| 219 |
-
fp16_backend='apex',
|
| 220 |
output_dir='seq2seq_model_printed/',
|
| 221 |
logging_strategy='epoch',
|
| 222 |
save_strategy='epoch',
|
|
|
|
| 214 |
evaluation_strategy='epoch',
|
| 215 |
per_device_train_batch_size=TrainingConfig.BATCH_SIZE,
|
| 216 |
per_device_eval_batch_size=TrainingConfig.BATCH_SIZE,
|
| 217 |
+
fp16=False,
|
| 218 |
+
# fp16_full_eval=True,
|
| 219 |
+
# fp16_backend='apex',
|
| 220 |
output_dir='seq2seq_model_printed/',
|
| 221 |
logging_strategy='epoch',
|
| 222 |
save_strategy='epoch',
|