Merge pull request #5 from taokong/master

fix mmcv version to resolve issue #4

Merge pull request #5 from taokong/master
fix mmcv version to resolve issue #4
925cc7cf · Xinlong Wang · GitHub · 6539835d · 83363a9a · 925cc7cf
Unverified Commit 925cc7cf authored Apr 01, 2020 by Xinlong Wang Committed by GitHub Apr 01, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

mmdet/apis/train.py mmdet/apis/train.py +4 -1

requirements/runtime.txt requirements/runtime.txt +1 -1

No files found.
--- a/mmdet/apis/train.py
+++ b/mmdet/apis/train.py
@@ -205,7 +205,10 @@ def _dist_train(model,
        for ds in dataset
    ]
    # 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
    optimizer = build_optimizer(model, cfg.optimizer)

--- a/requirements/runtime.txt
+++ b/requirements/runtime.txt
 matplotlib
-mmcv>=0.2.15
+mmcv>=0.3.1
 numpy
 scipy
 # need older pillow until torchvision is fixed