fix undesired assignment to pretrained key

open-mmlab · Aug 10, 2021 · c38a4d9 · c38a4d9
1 parent 084f8b2
commit c38a4d9
Showing 1 changed file with 2 additions and 1 deletion.
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: