perf(parser): 根据学习率线性缩放规则，分布式训练时仅改变学习率的大小

ZJCV · Nov 5, 2020 · d518e3d · d518e3d
1 parent ea7b99f
commit d518e3d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tsn/util/parser.py b/tsn/util/parser.py
@@ -135,8 +135,6 @@ def load_train_config(args):
     num_gpus = cfg.NUM_GPUS
     if num_gpus > 1:
         cfg.OPTIMIZER.LR *= num_gpus
-        cfg.OPTIMIZER.WEIGHT_DECAY *= num_gpus
-        cfg.LR_SCHEDULER.COSINE_ANNEALING_LR.MINIMAL_LR *= num_gpus
 
     cfg.freeze()