Spaces:

triopood
/

testocr

Sleeping

triopood commited on Jan 25

Commit

f7def6a

•

1 Parent(s): 5460ac0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -214,7 +214,7 @@ training_args = Seq2SeqTrainingArguments(
     evaluation_strategy='epoch',
     per_device_train_batch_size=TrainingConfig.BATCH_SIZE,
     per_device_eval_batch_size=TrainingConfig.BATCH_SIZE,
-    INT8=True,
     # fp16_full_eval=True,
     # fp16_backend='apex',
     output_dir='seq2seq_model_printed/',

     evaluation_strategy='epoch',
     per_device_train_batch_size=TrainingConfig.BATCH_SIZE,
     per_device_eval_batch_size=TrainingConfig.BATCH_SIZE,
+    fp16=False,
     # fp16_full_eval=True,
     # fp16_backend='apex',
     output_dir='seq2seq_model_printed/',