Kevin Fink commited on
Commit
733b7e9
·
1 Parent(s): 2a42237
Files changed (1) hide show
  1. app.py +2 -2
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=210,
88
  #push_to_hub=True,
89
  hub_model_id=hub_id.strip(),
90
  fp16=True,
91
  #lr_scheduler_type='cosine',
92
- save_steps=210, # Save checkpoint every 500 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