Spaces:

nroggendorff
/

train-llama

Paused

nroggendorff commited on Jul 18, 2024

Commit

7fe1c0a

verified ·

1 Parent(s): 5da7a55

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,7 +98,7 @@ print(dataset['text'][2])
 args = TrainingArguments(
     output_dir="mayo",
-    num_train_epochs=8,
     per_device_train_batch_size=16,
     gradient_accumulation_steps=4,
     learning_rate=1e-5,

 args = TrainingArguments(
     output_dir="mayo",
+    num_train_epochs=4,
     per_device_train_batch_size=16,
     gradient_accumulation_steps=4,
     learning_rate=1e-5,