Kevin Fink
commited on
Commit
·
801dbf0
1
Parent(s):
f0b3f80
deve
Browse files
app.py
CHANGED
|
@@ -227,7 +227,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 227 |
)
|
| 228 |
|
| 229 |
# Fine-tune the model
|
| 230 |
-
trainer.
|
| 231 |
#if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
|
| 232 |
#train_result = trainer.train(resume_from_checkpoint=True)
|
| 233 |
#else:
|
|
|
|
| 227 |
)
|
| 228 |
|
| 229 |
# Fine-tune the model
|
| 230 |
+
trainer.train()
|
| 231 |
#if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
|
| 232 |
#train_result = trainer.train(resume_from_checkpoint=True)
|
| 233 |
#else:
|