fix res5head config (#2138)

6d92ef31 · Feng Ni · GitHub · 87ece582 · 6d92ef31 · 6d92ef31
2 changed file
--- a/dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml
+++ b/dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml
@@ -10,5 +10,5 @@ ResNet:
  depth: 101
  norm_type: bn
  freeze_at: 0
-  return_idx: [0,1,2,3]
+  return_idx: [2]
-  num_stages: 4
+  num_stages: 3
--- a/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml
+++ b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml
@@ -10,5 +10,5 @@ ResNet:
  variant: d
  norm_type: bn
  freeze_at: 0
-  return_idx: [0,1,2,3]
+  return_idx: [2]
-  num_stages: 4
+  num_stages: 3