update cpu training and testing

mmfewshot/__init__.py

@@ -32,7 +32,7 @@ assert (digit_version(mmcv_minimum_version) <= mmcv_version
     f'Please install mmcv>={mmcv_minimum_version}, <={mmcv_maximum_version}.'
 
 mmdet_minimum_version = '2.16.0'
-mmdet_maximum_version = '2.21.0'
+mmdet_maximum_version = '2.23.0'
 mmdet_version = digit_version(mmdet.__version__)
 
 

mmfewshot/detection/apis/train.py

@@ -55,6 +55,7 @@ def train_detector(model: nn.Module,
             broadcast_buffers=False,
             find_unused_parameters=find_unused_parameters)
     else:
+        # Please use MMCV >= 1.4.4 for CPU training!
         model = MMDataParallel(model, device_ids=cfg.gpu_ids)
 
     # build runner

tools/classification/test.py

@@ -70,7 +70,13 @@ def parse_args():
         '--gpu-ids',
         type=int,
         nargs='+',
-        help='ids of gpus to use '
+        help='(Deprecated, please use --gpu-id) ids of gpus to use '
+        '(only applicable to non-distributed testing)')
+    parser.add_argument(
+        '--gpu-id',
+        type=int,
+        default=0,
+        help='id of gpu to use '
         '(only applicable to non-distributed testing)')
     parser.add_argument(
         '--show_task_results',

tools/detection/test.py

@@ -195,6 +195,7 @@ def main():
             shuffle=False)
 
     if not distributed:
+        # Please use MMCV >= 1.4.4 for CPU testing!
         model = MMDataParallel(model, device_ids=cfg.gpu_ids)
         show_kwargs = dict(show_score_thr=args.show_score_thr)
         if cfg.data.get('model_init', None) is not None:

tools/detection/train.py

@@ -142,6 +142,7 @@ def main():
     # init distributed env first, since logger depends on the dist info.
     if args.launcher == 'none':
         distributed = False
+        rank = 0
     else:
         distributed = True
         init_dist(args.launcher, **cfg.dist_params)