Kevin Fink
commited on
Commit
·
297c33f
1
Parent(s):
08c19d5
dev
Browse files
app.py
CHANGED
|
@@ -73,7 +73,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 73 |
logging_steps=10,
|
| 74 |
#push_to_hub=True,
|
| 75 |
hub_model_id=hub_id.strip(),
|
| 76 |
-
|
| 77 |
#lr_scheduler_type='cosine',
|
| 78 |
save_steps=100, # Save checkpoint every 500 steps
|
| 79 |
save_total_limit=3,
|
|
|
|
| 73 |
logging_steps=10,
|
| 74 |
#push_to_hub=True,
|
| 75 |
hub_model_id=hub_id.strip(),
|
| 76 |
+
fp16=True,
|
| 77 |
#lr_scheduler_type='cosine',
|
| 78 |
save_steps=100, # Save checkpoint every 500 steps
|
| 79 |
save_total_limit=3,
|