Kevin Fink
commited on
Commit
·
22093a9
1
Parent(s):
cb7d042
init
Browse files
app.py
CHANGED
|
@@ -31,7 +31,8 @@ def fine_tune_model(model_name, dataset_name, hub_id, num_epochs, batch_size, lr
|
|
| 31 |
# Set training arguments
|
| 32 |
training_args = TrainingArguments(
|
| 33 |
output_dir='./results',
|
| 34 |
-
|
|
|
|
| 35 |
learning_rate=lr**-5,
|
| 36 |
per_device_train_batch_size=batch_size,
|
| 37 |
per_device_eval_batch_size=batch_size,
|
|
|
|
| 31 |
# Set training arguments
|
| 32 |
training_args = TrainingArguments(
|
| 33 |
output_dir='./results',
|
| 34 |
+
eval_strategy="epoch",
|
| 35 |
+
save_strategy='epoch',
|
| 36 |
learning_rate=lr**-5,
|
| 37 |
per_device_train_batch_size=batch_size,
|
| 38 |
per_device_eval_batch_size=batch_size,
|