From ace0e41c60e268137e3ed5db3155ac2a41bbbf52 Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Tue, 31 Mar 2020 22:54:16 +0800 Subject: [PATCH 1/6] fix mmcv version to resolve issue #4 --- requirements/runtime.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 898c0e2a..916f1396 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,5 +1,5 @@ matplotlib -mmcv>=0.2.15 +mmcv>=0.2.15,<=0.4.0 numpy scipy # need older pillow until torchvision is fixed From 394593871313c0cdc7165036404b175f0f83856c Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Tue, 31 Mar 2020 23:55:38 +0800 Subject: [PATCH 2/6] Update runtime.txt --- requirements/runtime.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 916f1396..0d017878 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,5 +1,5 @@ matplotlib -mmcv>=0.2.15,<=0.4.0 +mmcv==0.2.16 numpy scipy # need older pillow until torchvision is fixed From 166f4d8c53653789c62a2a4c46e5bb94f8d781b0 Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Wed, 1 Apr 2020 00:21:42 +0800 Subject: [PATCH 3/6] Update INSTALL.md --- docs/INSTALL.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/INSTALL.md b/docs/INSTALL.md index 5a35ef6a..d2888380 100644 --- a/docs/INSTALL.md +++ b/docs/INSTALL.md @@ -8,7 +8,7 @@ - CUDA 9.0 or higher - NCCL 2 - GCC 4.9 or higher -- [mmcv](https://github.com/open-mmlab/mmcv) +- [mmcv 0.2.16](https://github.com/open-mmlab/mmcv/tree/v0.2.16) We have tested the following versions of OS and softwares: From 1f0be8e7063439537d497afb02a73b825abb6a97 Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Wed, 1 Apr 2020 11:20:06 +0800 Subject: [PATCH 4/6] fix mmcv version --- requirements/runtime.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 0d017878..a230edfc 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,5 +1,5 @@ matplotlib -mmcv==0.2.16 +mmcv>=0.3.1 numpy scipy # need older pillow until torchvision is fixed From 1e79dac4272f2f5cc42442718f74707ded8868b0 Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Wed, 1 Apr 2020 11:20:49 +0800 Subject: [PATCH 5/6] Update INSTALL.md --- docs/INSTALL.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/INSTALL.md b/docs/INSTALL.md index d2888380..5a35ef6a 100644 --- a/docs/INSTALL.md +++ b/docs/INSTALL.md @@ -8,7 +8,7 @@ - CUDA 9.0 or higher - NCCL 2 - GCC 4.9 or higher -- [mmcv 0.2.16](https://github.com/open-mmlab/mmcv/tree/v0.2.16) +- [mmcv](https://github.com/open-mmlab/mmcv) We have tested the following versions of OS and softwares: From 83363a9a9f6c02dda105ec0686f3d6426b13693d Mon Sep 17 00:00:00 2001 From: Tao Kong Date: Wed, 1 Apr 2020 11:22:35 +0800 Subject: [PATCH 6/6] fix mmcv version --- mmdet/apis/train.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mmdet/apis/train.py b/mmdet/apis/train.py index 97c0dc69..9d6f5bb8 100644 --- 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)