mmtune/mm/tasks/mmdet.py

@@ -60,14 +60,13 @@ def parse_args(self, args: Sequence[str]) -> argparse.Namespace:
             'like key="[a,b]" or key=a,b It also allows nested list/tuple '
             'values, e.g. key="[(a,b),(c,d)]" Note that the quotation marks '
             'are necessary and that no white space is allowed.')
-        parser.add_argument('--local_rank', type=int, default=0)
         parser.add_argument(
             '--auto-scale-lr',
             action='store_true',
             help='enable automatically scaling LR.')
         args = parser.parse_args(args)
         if 'LOCAL_RANK' not in os.environ:
-            os.environ['LOCAL_RANK'] = str(args.local_rank)
+            os.environ['LOCAL_RANK'] = str(dist.get_rank())
         return args
 
     def build_model(self,
@@ -106,7 +105,6 @@ def run(self, *args, **kwargs):
         cfg = Config.fromfile(args.config)
         if args.cfg_options is not None:
             cfg.merge_from_dict(args.cfg_options)
-        cfg.device = ''
 
         # set cudnn_benchmark
         if cfg.get('cudnn_benchmark', False):