Charlie81
/

LoRE

Model card Files Files and versions

Metrics Training metrics Community

Charlie81 commited on Jul 11, 2025

Commit

85d731b

·

1 Parent(s): d55ddf7

batch 3

Files changed (1) hide show

scripts/train.py +1 -1

scripts/train.py CHANGED Viewed

@@ -70,7 +70,7 @@ def main():
     # Training arguments
     training_args = TrainingArguments(
         output_dir="./output",
-        per_device_train_batch_size=2,
         gradient_accumulation_steps=8,
         learning_rate=1e-4,
         num_train_epochs=1,

     # Training arguments
     training_args = TrainingArguments(
         output_dir="./output",
+        per_device_train_batch_size=3
         gradient_accumulation_steps=8,
         learning_rate=1e-4,
         num_train_epochs=1,