From a72cbf94ec4a28aee1bb7e23a543f0c7e6afca95 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Sat, 17 Apr 2021 00:32:29 +0800
Subject: [PATCH] update

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index f8bc737..b7f159b 100644
--- a/train.py
+++ b/train.py
@@ -194,7 +194,7 @@ if __name__ == "__main__":
         help = 'imageNet_Pretrain') # 初始化学习率
     parser.add_argument('--fintune_model', type=str, default = 'None',
         help = 'fintune_model') # fintune model
-    parser.add_argument('--loss_define', type=str, default = 'mse_loss',
+    parser.add_argument('--loss_define', type=str, default = 'adaptive_wing_loss',
         help = 'define_loss : wing_loss, mse_loss ,adaptive_wing_loss') # 损失函数定义
     parser.add_argument('--init_lr', type=float, default = 1e-3,
         help = 'init learning Rate') # 初始化学习率
-- 
GitLab