diff --git a/configs/faster_rcnn_r101_1x.yml b/configs/faster_rcnn_r101_1x.yml index d940477567c447aad35fa231b38ebe7b2059c748..b04a9fc42eb5328eea0292c222d0e724f9a35146 100644 --- a/configs/faster_rcnn_r101_1x.yml +++ b/configs/faster_rcnn_r101_1x.yml @@ -25,6 +25,7 @@ ResNet: freeze_at: 2 ResNetC5: + depth: 101 norm_type: affine_channel RPNHead: diff --git a/configs/faster_rcnn_r50_1x.yml b/configs/faster_rcnn_r50_1x.yml index 10002a176c8f5352414f195ae1dab3f2f94cd5e8..9ec90c674d1a5608cef205aac0e2820db6a625ca 100644 --- a/configs/faster_rcnn_r50_1x.yml +++ b/configs/faster_rcnn_r50_1x.yml @@ -25,6 +25,7 @@ ResNet: freeze_at: 2 ResNetC5: + depth: 50 norm_type: affine_channel RPNHead: diff --git a/configs/faster_rcnn_r50_2x.yml b/configs/faster_rcnn_r50_2x.yml index 0e9686b1a7ba8f144d8903bed74fdfb546dc7779..c28bcc25f9781cb84f03eeb782011bbb7518bb47 100644 --- a/configs/faster_rcnn_r50_2x.yml +++ b/configs/faster_rcnn_r50_2x.yml @@ -25,6 +25,7 @@ ResNet: freeze_at: 2 ResNetC5: + depth: 50 norm_type: affine_channel RPNHead: diff --git a/configs/faster_rcnn_r50_vd_1x.yml b/configs/faster_rcnn_r50_vd_1x.yml index 528760d94653fe3b025cc2bd29452a23df7a183c..5941a03777f480e203e0e046da7aa4255d06da7b 100644 --- a/configs/faster_rcnn_r50_vd_1x.yml +++ b/configs/faster_rcnn_r50_vd_1x.yml @@ -26,6 +26,7 @@ ResNet: variant: d ResNetC5: + depth: 50 norm_type: affine_channel variant: d diff --git a/configs/mask_rcnn_r50_1x.yml b/configs/mask_rcnn_r50_1x.yml index 9938d289b49fdf5cf273c724663250d1d456f0a9..571be5fef110f857719a8630c341d1a81f4c00e4 100644 --- a/configs/mask_rcnn_r50_1x.yml +++ b/configs/mask_rcnn_r50_1x.yml @@ -28,6 +28,7 @@ ResNet: freeze_at: 2 ResNetC5: + depth: 50 norm_type: affine_channel RPNHead: diff --git a/configs/mask_rcnn_r50_2x.yml b/configs/mask_rcnn_r50_2x.yml index 97009098d000b343851fcd1bfc9750ed59e3ef5a..2c3b59b220f1e4aeaeee04c879c09fec8576e0f5 100644 --- a/configs/mask_rcnn_r50_2x.yml +++ b/configs/mask_rcnn_r50_2x.yml @@ -29,6 +29,7 @@ ResNet: freeze_at: 2 ResNetC5: + depth: 50 norm_type: affine_channel RPNHead: diff --git a/ppdet/utils/checkpoint.py b/ppdet/utils/checkpoint.py index 4a20ef7256849bb012de6f7890291131152b3a73..b5bbc792c4034e3d3b09b5b4e14b771e0d7836ad 100644 --- a/ppdet/utils/checkpoint.py +++ b/ppdet/utils/checkpoint.py @@ -183,7 +183,8 @@ def load_and_fusebn(exe, prog, path): [scale_name, bias_name, mean_name, variance_name]) if not bn_in_path: - raise ValueError("The model in path {} has not params of batch norm.") + raise ValueError("There is no params of batch norm in model {}.".format( + path)) # load running mean and running variance on cpu place into global scope. place = fluid.CPUPlace()