Fix open-mmlab#416: Fix undesired assignment to "pretrained" in test.…

…py (open-mmlab#418) * fix ocr linting * fix undesired assignment to pretrained key
gaotongxiao · Aug 10, 2021 · b373201 · b373201
1 parent c068ccb
commit b373201
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mmocr/utils/ocr.py b/mmocr/utils/ocr.py
@@ -275,8 +275,7 @@ def __init__(self,
             },
             'RobustScanner': {
                 'config': 'robust_scanner/robustscanner_r31_academic.py',
-                'ckpt':
-                'robustscanner/robustscanner_r31_academic-5f05874f.pth'
+                'ckpt': 'robustscanner/robustscanner_r31_academic-5f05874f.pth'
             },
             'SEG': {
                 'config': 'seg/seg_r31_1by16_fpnocr_academic.py',

diff --git a/tools/test.py b/tools/test.py
@@ -127,7 +127,8 @@ def main():
     # set cudnn_benchmark
     if cfg.get('cudnn_benchmark', False):
         torch.backends.cudnn.benchmark = True
-    cfg.model.pretrained = None
+    if cfg.model.get('pretrained'):
+        cfg.model.pretrained = None
     if cfg.model.get('neck'):
         if isinstance(cfg.model.neck, list):
             for neck_cfg in cfg.model.neck: