Update train_cls.py

FuNz-0 · Sep 1, 2024 · 0f86ce0 · 0f86ce0
1 parent 0315d6d
commit 0f86ce0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_cls.py b/train_cls.py
@@ -230,10 +230,10 @@ def get_args():
 
     # prompt tuning hyper-parameter
     parser.add_argument("--n_ctx", type=int, default=4)
-    parser.add_argument("--n_ctx_ab", type=int, default=100)
+    parser.add_argument("--n_ctx_ab", type=int, default=1)
     parser.add_argument("--n_pro", type=int, default=3)
     parser.add_argument("--n_pro_ab", type=int, default=4)
-    parser.add_argument("--Epoch", type=int, default=1)
+    parser.add_argument("--Epoch", type=int, default=100)
 
     # optimizer
     parser.add_argument("--lr", type=float, default=0.002)