修改参数

train/grid_search_cosface.py

@@ -41,11 +41,17 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 # ==================== 网格搜索空间定义 ====================
-GRID_PARAMS = {
-    'dish': {
+"""
+'dish': {
         's': [56.0, 60.0, 64.0, 68.0],  # scale参数
         'm': [0.32, 0.35, 0.38, 0.40],  # margin参数
     },
+"""
+GRID_PARAMS = {
+    'dish': {
+        's': [56.0, 60.0],  # scale参数
+        'm': [0.32,  0.40],  # margin参数
+    },
     'whole_ingredient': {
         's': [56.0, 60.0, 64.0, 68.0],
         'm': [0.32, 0.35, 0.38, 0.40],
@@ -354,8 +360,8 @@ def grid_search_main(
 if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser(description='CosFace超参数网格搜索')
-    parser.add_argument('--task', choices=list(TASKS.keys()), default='dish', help='任务名称')
-    # parser.add_argument('--task', choices=list(TASKS.keys()), default='whole_ingredient', help='任务名称')
+    # parser.add_argument('--task', choices=list(TASKS.keys()), default='dish', help='任务名称')
+    parser.add_argument('--task', choices=list(TASKS.keys()), default='whole_ingredient', help='任务名称')
     parser.add_argument('--max_configs', type=int, default=None, help='最大配置数（用于测试）')
     parser.add_argument('--epochs', type=int, default=100, help='每个配置的最大训练轮数')
     parser.add_argument('--patience', type=int, default=10, help='早停容忍轮数')