Spaces:

amine-araich
/

controlled-text-summarization

Runtime error

amine-araich commited on Mar 14

Commit

36f1333

verified ·

1 Parent(s): a53c147

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -21,7 +21,7 @@ def fine_tune_model(model, tokenizer, dataset, output_dir="./summarization_model
         per_device_eval_batch_size=4,
         gradient_accumulation_steps=4,
         learning_rate=5e-5,
-        num_train_epochs=7,
         save_strategy="epoch",
         eval_strategy="epoch",
         load_best_model_at_end=True,

         per_device_eval_batch_size=4,
         gradient_accumulation_steps=4,
         learning_rate=5e-5,
+        num_train_epochs=6,
         save_strategy="epoch",
         eval_strategy="epoch",
         load_best_model_at_end=True,