From fc594181ed59884d6022d0b84e2c0cd2ac5919ac Mon Sep 17 00:00:00 2001 From: zhangpu <1250681871@qq.com> Date: Tue, 11 Nov 2025 10:42:00 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9margin=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- train/train_cosface_embedding.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/train/train_cosface_embedding.py b/train/train_cosface_embedding.py index be92c5c..5748dcb 100644 --- a/train/train_cosface_embedding.py +++ b/train/train_cosface_embedding.py @@ -52,8 +52,10 @@ TASKS = { batch_size=32, lr=5e-4, aug_strength='medium', - cosface_s=60.0, # 菜品分类：类内差异大，使用中等scale - cosface_m=0.32, # 较小margin，适应类内多样性（不同做法、角度） + # cosface_s=60.0, # 菜品分类：类内差异大，使用中等scale + cosface_s=64.0, # 菜品分类：类内差异大，使用中等scale + # cosface_m=0.32, # 较小margin，适应类内多样性（不同做法、角度） + cosface_m=0.40, # 较小margin，适应类内多样性（不同做法、角度） ), 'whole_ingredient': TaskConfig( name='WholeIngredientRecognition', @@ -406,7 +408,8 @@ if __name__ == '__main__': # 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) + # parser.add_argument('--epochs', type=int, default=60) + parser.add_argument('--epochs', type=int, default=100) parser.add_argument('--patience', type=int, default=10, help='早停容忍轮数') parser.add_argument('--min_delta', type=float, default=0.001, help='早停最小改善阈值') parser.add_argument('--min_epochs', type=int, default=25, help='最小训练轮数（早停保护）')