Spaces:

wangjin2000
/

ESM2Bind

Paused

wangjin2000 commited on Jun 28, 2024

Commit

b9700ce

verified ·

1 Parent(s): e474658

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -91,7 +91,7 @@ def train_function_no_sweeps(base_model_path):   #, train_dataset, test_dataset)
         "lr": 5.701568055793089e-04,
         "lr_scheduler_type": "cosine",
         "max_grad_norm": 0.5,
-        "num_train_epochs": 3,
         "per_device_train_batch_size": 12,
         "r": 2,
         "weight_decay": 0.2,

         "lr": 5.701568055793089e-04,
         "lr_scheduler_type": "cosine",
         "max_grad_norm": 0.5,
+        "num_train_epochs": 2,  #3, jw 20240628
         "per_device_train_batch_size": 12,
         "r": 2,
         "weight_decay": 0.2,