mod

2025-02-25 13:54:56 +01:00 · 2025-02-25 13:54:56 +01:00 · 90f3d66723
parent b3d1e0295a
commit 90f3d66723
1 changed files with 0 additions and 1 deletions
--- a/hft.py
+++ b/hft.py
@ -170,7 +170,6 @@ model.gradient_checkpointing_enable()
 training_args = TrainingArguments(
    output_dir="./results",
    num_train_epochs=3,
-    per_device_train_batch_size=2,
    gradient_accumulation_steps=4,
    learning_rate=2e-5,
    fp16=True,