Skip to content
Snippets Groups Projects
Unverified Commit 7835f640 authored by boli's avatar boli Committed by GitHub
Browse files

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

  Remove duplicate operations
parents 280c1255 6d056217
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment