[Fix] :fix pretraied error (open-mmlab#5550)

qixiong-wang · Jul 6, 2021 · 7b5a58b · 7b5a58b
1 parent 221fe0e
commit 7b5a58b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/configs/cascade_rcnn/cascade_mask_rcnn_r50_caffe_fpn_1x_coco.py b/configs/cascade_rcnn/cascade_mask_rcnn_r50_caffe_fpn_1x_coco.py
@@ -2,10 +2,12 @@
 
 model = dict(
     backbone=dict(
-        norm_cfg=dict(requires_grad=False), norm_eval=True, style='caffe'),
-    init_cfg=dict(
-        type='Pretrained',
-        checkpoint='open-mmlab://detectron2/resnet50_caffe'))
+        norm_cfg=dict(requires_grad=False),
+        norm_eval=True,
+        style='caffe',
+        init_cfg=dict(
+            type='Pretrained',
+            checkpoint='open-mmlab://detectron2/resnet50_caffe')))
 img_norm_cfg = dict(
     mean=[103.530, 116.280, 123.675], std=[1.0, 1.0, 1.0], to_rgb=False)
 train_pipeline = [