eibeel
/

mixtral_FineTuning_TFG

eibeel commited on May 23, 2024

Commit

d1e4843

verified ·

1 Parent(s): 829f82f

Update MIXTRAL_Fine-Tuning.py

Files changed (1) hide show

MIXTRAL_Fine-Tuning.py CHANGED Viewed

@@ -126,7 +126,7 @@ training_arguments = TrainingArguments(
         per_device_eval_batch_size=4,
         log_level="debug",
         save_steps=1000,
-        logging_steps=100,
         learning_rate=2e-4,
         eval_steps=500,
         max_steps=-1,

         per_device_eval_batch_size=4,
         log_level="debug",
         save_steps=1000,
+        logging_steps=logging_steps,
         learning_rate=2e-4,
         eval_steps=500,
         max_steps=-1,