From b4b7a249e329ff2ce3be8b85ea3964d2f7799f28 Mon Sep 17 00:00:00 2001 From: randaller Date: Sun, 19 Mar 2023 13:33:59 +0300 Subject: [PATCH] Update hf-training-example.py --- hf-training-example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)