Update main_gs.py

ShengcaiLiao · Apr 1, 2021 · 9217f41 · 9217f41
1 parent 8e76e9d
commit 9217f41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main_gs.py b/main_gs.py
@@ -370,8 +370,8 @@ def main(args):
                         help="the path to the image data")
     parser.add_argument('--exp-dir', type=str, metavar='PATH', default=osp.join(working_dir, 'Exp'),
                         help="the path to the output directory")
-    parser.add_argument('--method', type=str, default='QAConv', help="method name for the output directory")
-    parser.add_argument('--sub_method', type=str, default='res50_layer3',
+    parser.add_argument('--method', type=str, default='QAConv50_IBNb_GS', help="method name for the output directory")
+    parser.add_argument('--sub_method', type=str, default='res50-layer3-f64_pre1_clip512_lr3_bs64-k4_ep15s10',
                         help="sub method name for the output directory")
     parser.add_argument('--save_score', default=False, action='store_true',
                         help="save the matching score or not, default: False")