Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 10, 2024

Commit

a520e10

1 Parent(s): b2bf87a

dev

Files changed (1) hide show

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
             metric_for_best_model="loss",
             greater_is_better=True,
             logging_dir='/data/logs',
-            logging_steps=100,
             #push_to_hub=True,
             hub_model_id=hub_id.strip(),
             fp16=True,

             metric_for_best_model="loss",
             greater_is_better=True,
             logging_dir='/data/logs',
+            logging_steps=200,
             #push_to_hub=True,
             hub_model_id=hub_id.strip(),
             fp16=True,