wangjin2000 commited on
Commit
2aa5124
·
verified ·
1 Parent(s): e23f23b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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,