diff --git a/hf-training-example.py b/hf-training-example.py index cd518bb..2afb9d4 100644 --- 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)