triopood commited on
Commit
4c08581
1 Parent(s): 8c7e03f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=True,
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',