处理embedding的请求时长。

train/train_cosface_embedding.py

@@ -68,8 +68,8 @@ TASKS = {
         batch_size=64,
         lr=8e-4,
         aug_strength='medium',
-        cosface_s=64.0,  # 完整食材：类间区分度高，使用标准scale
-        cosface_m=0.38,  # 较大margin，强化类间分离（番茄vs土豆差异明显）
+        cosface_s=68.0,  # 完整食材：类间区分度高，使用标准scale
+        cosface_m=0.35,  # 较大margin，强化类间分离（番茄vs土豆差异明显）
     ),
     'processed_ingredient': TaskConfig(
         name='ProcessedIngredientRecognition',
@@ -422,8 +422,8 @@ def main(task_key: str = 'dish', s: Optional[float] = None, m: Optional[float] =
 if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='dish')
-    # parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='whole_ingredient')
+    # parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='dish')
+    parser.add_argument('task', choices=list(TASKS.keys()), nargs='?', default='whole_ingredient')
     parser.add_argument('--s', type=float, default=None, help='CosFace scale factor (默认使用任务配置值)')
     parser.add_argument('--m', type=float, default=None, help='CosFace margin (默认使用任务配置值)')
     # parser.add_argument('--epochs', type=int, default=60)