Skip to content

GitLab

Unverified Commit 925cc7cf authored by Xinlong Wang's avatar Xinlong Wang Committed by GitHub
Browse files

Merge pull request #5 from taokong/master 

fix mmcv version to resolve issue #4
parents 6539835d 83363a9a
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
mmdet/apis/train.py
View file @ 925cc7cf
...@@ -205,7 +205,10 @@ def _dist_train(model, ...@@ -205,7 +205,10 @@ def _dist_train(model,
for ds in dataset for ds in dataset
] ]
# put model on gpus # put model on gpus
model = MMDistributedDataParallel(model.cuda()) # model = MMDistributedDataParallel(model.cuda())
model = MMDistributedDataParallel(model.cuda(),
device_ids=[torch.cuda.current_device()],
broadcast_buffers=False)
# build runner # build runner
optimizer = build_optimizer(model, cfg.optimizer) optimizer = build_optimizer(model, cfg.optimizer)
......
requirements/runtime.txt
View file @ 925cc7cf
matplotlib matplotlib
mmcv>=0.2.15 mmcv>=0.3.1
numpy numpy
scipy scipy
# need older pillow until torchvision is fixed # need older pillow until torchvision is fixed
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment