diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml index f286c35c14255c43dca7f66d2ed2dfd30984201a..93c3deace5a34bd7f556e100d851d397c5c9b585 100644 --- 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] - num_stages: 4 + return_idx: [2] + num_stages: 3 diff --git a/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml b/dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml index de68619fedf2ed89684fbec2e995dd6274189cbe..f0a0d8216ec6cd3b92bcf3064c3e36c938a58dc5 100644 --- 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] - num_stages: 4 + return_idx: [2] + num_stages: 3