Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 8, 2024

Commit

406797f

1 Parent(s): f06d0fa

dev

Files changed (1) hide show

app.py CHANGED Viewed

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

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