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',
|