Support CPU training (open-mmlab#404)

* Update train.py * Update __init__.py
shuyangsun · Jan 26, 2022 · 0d9febd · 0d9febd
1 parent bb0a643
commit 0d9febd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mmtrack/__init__.py b/mmtrack/__init__.py
@@ -7,10 +7,10 @@
 
 from .version import __version__, version_info
 
-MMCV_MIN = '1.3.8'
+MMCV_MIN = '1.3.17'
 MMCV_MAX = '1.5.0'
 
-MMDET_MIN = '2.14.0'
+MMDET_MIN = '2.19.1'
 
 
 def digit_version(version_str: str, length: int = 4):

diff --git a/mmtrack/apis/train.py b/mmtrack/apis/train.py
@@ -109,7 +109,7 @@ def train_model(model,
             find_unused_parameters=find_unused_parameters)
     else:
         model = MMDataParallel(
-            model.cuda(cfg.gpu_ids[0]), device_ids=cfg.gpu_ids)
+            model, device_ids=cfg.gpu_ids)
 
     # build runner
     optimizer = build_optimizer(model, cfg.optimizer)