Kevin Fink
commited on
Commit
·
733b7e9
1
Parent(s):
2a42237
deve
Browse files
app.py
CHANGED
@@ -84,12 +84,12 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
84 |
metric_for_best_model="loss",
|
85 |
greater_is_better=True,
|
86 |
logging_dir='/data/logs',
|
87 |
-
logging_steps=
|
88 |
#push_to_hub=True,
|
89 |
hub_model_id=hub_id.strip(),
|
90 |
fp16=True,
|
91 |
#lr_scheduler_type='cosine',
|
92 |
-
save_steps=
|
93 |
save_total_limit=3,
|
94 |
)
|
95 |
# Check if a checkpoint exists and load it
|
|
|
84 |
metric_for_best_model="loss",
|
85 |
greater_is_better=True,
|
86 |
logging_dir='/data/logs',
|
87 |
+
logging_steps=200,
|
88 |
#push_to_hub=True,
|
89 |
hub_model_id=hub_id.strip(),
|
90 |
fp16=True,
|
91 |
#lr_scheduler_type='cosine',
|
92 |
+
save_steps=200, # Save checkpoint every 500 steps
|
93 |
save_total_limit=3,
|
94 |
)
|
95 |
# Check if a checkpoint exists and load it
|