Merge pull request #121 from 103yiran/103yiran-train

  Remove duplicate operations

tools/train.py

@@ -273,7 +273,6 @@ def main():
 
     # create model
     model = ModelBuilder().cuda().train()
-    dist_model = DistModule(model)
 
     # load pretrained backbone weights
     if cfg.BACKBONE.PRETRAINED:
@@ -291,7 +290,7 @@ def main():
     train_loader = build_data_loader()
 
     # build optimizer and lr_scheduler
-    optimizer, lr_scheduler = build_opt_lr(dist_model.module,
+    optimizer, lr_scheduler = build_opt_lr(model,
                                            cfg.TRAIN.START_EPOCH)
 
     # resume training