diff --git a/train.py b/train.py
index f8bc737af759d9d24f2ddc0471c46601e6c1b1ea..b7f159bae81aa18fb84abb70a3666c527a7addfd 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') # 初始化学习率