From 6d92ef310233184690ed5db2e1a52f0271618568 Mon Sep 17 00:00:00 2001 From: Feng Ni Date: Sat, 30 Jan 2021 19:14:41 +0800 Subject: [PATCH] fix res5head config (#2138) --- dygraph/configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml | 4 ++-- dygraph/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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 f286c35c1..93c3deace 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 de68619fe..f0a0d8216 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 -- GitLab