Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -75,9 +75,10 @@ def Finetune(base_model_path): #, train_dataset, test_dataset):
|
|
| 75 |
model_name_base = base_model_path.split("/")[1]
|
| 76 |
timestamp = datetime.now().strftime('%Y-%m-%d_%H')
|
| 77 |
lr = 0.0007984276816171436
|
|
|
|
| 78 |
|
| 79 |
training_args = TrainingArguments(
|
| 80 |
-
output_dir=f"{model_name_base}_PPI_{timestamp}",
|
| 81 |
num_train_epochs = 5,
|
| 82 |
per_device_train_batch_size = 2,
|
| 83 |
per_device_eval_batch_size = 16,
|
|
|
|
| 75 |
model_name_base = base_model_path.split("/")[1]
|
| 76 |
timestamp = datetime.now().strftime('%Y-%m-%d_%H')
|
| 77 |
lr = 0.0007984276816171436
|
| 78 |
+
save_path = f"{model_name_base}_PPI_{timestamp}"
|
| 79 |
|
| 80 |
training_args = TrainingArguments(
|
| 81 |
+
output_dir=save_path, #f"{model_name_base}_PPI_{timestamp}",
|
| 82 |
num_train_epochs = 5,
|
| 83 |
per_device_train_batch_size = 2,
|
| 84 |
per_device_eval_batch_size = 16,
|