增加多模态模型开放集识别。

2025-11-13 11:56:51 +08:00
parent b4b2b18ccd
commit ced0e4f7ef
7 changed files with 628 additions and 21 deletions
@@ -354,8 +354,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='早停容忍轮数')