shape not same

[zxw223311]

RuntimeError: For 'load_param_into_net', rcnn_mask.fpn_mask.mask_conv6.weight in the argument 'net' should have the same shape as rcnn_mask.fpn_mask.mask_conv6.weight in the argument 'parameter_dict'. But got its shape (4, 256, 1, 1) in the argument 'net' and shape (81, 256, 1, 1) in the argument 'parameter_dict'.May you need to check whether the checkpoint you loaded is correct or the batch size and so on in the 'net' and 'parameter_dict' are same.

[zxw223311]

how to adjust shape,thanks