Merge 4e2cf3d into 1efdbda

open-mmlab · Nov 9, 2021 · 2159aae · 2159aae
2 parents 1efdbda + 4e2cf3d
commit 2159aae
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmocr/apis/inference.py b/mmocr/apis/inference.py
@@ -40,8 +40,7 @@ def init_detector(config, checkpoint=None, device='cuda:0', cfg_options=None):
     config.model.train_cfg = None
     model = build_detector(config.model, test_cfg=config.get('test_cfg'))
     if checkpoint is not None:
-        map_loc = 'cpu' if device == 'cpu' else None
-        checkpoint = load_checkpoint(model, checkpoint, map_location=map_loc)
+        checkpoint = load_checkpoint(model, checkpoint, map_location='cpu')
         if 'CLASSES' in checkpoint.get('meta', {}):
             model.CLASSES = checkpoint['meta']['CLASSES']
         else: