Update hf-training-example.py

3 years ago · b4b7a249e3
parent 0ff8e8dfe1
commit b4b7a249e3
1 changed files with 1 additions and 1 deletions
--- a/hf-training-example.py
+++ b/hf-training-example.py
@ -48,7 +48,7 @@ training_args = TrainingArguments(
    logging_dir='./logs',
    output_dir=OUTPUT_DIR,
    no_cuda=True,
-    bf16=True,
+    # bf16=True,
    per_device_eval_batch_size=1,
    per_device_train_batch_size=1)