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

Remove duplicate operations
STVIR · Dec 20, 2019 · 7835f64 · 7835f64
2 parents 280c125 + 6d05621
commit 7835f64
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tools/train.py b/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