Update app.py
Browse files
app.py
CHANGED
|
@@ -214,7 +214,7 @@ 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 |
-
|
| 218 |
# fp16_full_eval=True,
|
| 219 |
# fp16_backend='apex',
|
| 220 |
output_dir='seq2seq_model_printed/',
|
|
|
|
| 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/',
|