Kevin Fink commited on
Commit
cf05c78
·
1 Parent(s): e89a055
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -233,8 +233,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
233
  #)
234
 
235
  # Fine-tune the model
236
- if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
237
- train_result = trainer.train(resume_from_checkpoint=True)
238
  else:
239
  train_result = trainer.train()
240
  trainer.push_to_hub(commit_message="Training complete!")
 
233
  #)
234
 
235
  # Fine-tune the model
236
+ #if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
237
+ #train_result = trainer.train(resume_from_checkpoint=True)
238
  else:
239
  train_result = trainer.train()
240
  trainer.push_to_hub(commit_message="Training complete!")