Kevin Fink
commited on
Commit
·
63431bc
1
Parent(s):
8ba99ef
dev
Browse files
app.py
CHANGED
|
@@ -70,7 +70,6 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
|
|
| 70 |
training_args = TrainingArguments(
|
| 71 |
remove_unused_columns=False,
|
| 72 |
torch_empty_cache_steps=100,
|
| 73 |
-
overwrite_output_dir=True,
|
| 74 |
output_dir='/data/results',
|
| 75 |
eval_strategy="steps", # Change this to steps
|
| 76 |
save_strategy='steps',
|
|
|
|
| 70 |
training_args = TrainingArguments(
|
| 71 |
remove_unused_columns=False,
|
| 72 |
torch_empty_cache_steps=100,
|
|
|
|
| 73 |
output_dir='/data/results',
|
| 74 |
eval_strategy="steps", # Change this to steps
|
| 75 |
save_strategy='steps',
|