• Preston Parry's avatar
    Fixes bug with double (overwriting) RESNETS params (#488) · f40f2993
    Preston Parry authored
    There were two `RESNETS` sections, which overrode each other, leading to error messages like: 
    
    ```
    RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN:
    	size mismatch for backbone.fpn.fpn_inner1.weight: copying a param with shape torch.Size([256, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([1024, 256, 1, 1]).
    ...
    	size mismatch for roi_heads.mask.feature_extractor.mask_fcn1.weight: copying a param with shape torch.Size([256, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 1024, 3, 3]).
    ```
    
    This just combines them back into a single section, while maintaining all param values. That got the model running again for me.
    f40f2993
Name
Last commit
Last update
..
caffe2 Loading commit data...
cityscapes Loading commit data...
gn_baselines Loading commit data...
pascal_voc Loading commit data...
quick_schedules Loading commit data...
retinanet Loading commit data...
e2e_faster_rcnn_R_101_FPN_1x.yaml Loading commit data...
e2e_faster_rcnn_R_50_C4_1x.yaml Loading commit data...
e2e_faster_rcnn_R_50_FPN_1x.yaml Loading commit data...
e2e_faster_rcnn_X_101_32x8d_FPN_1x.yaml Loading commit data...
e2e_faster_rcnn_fbnet.yaml Loading commit data...
e2e_faster_rcnn_fbnet_600.yaml Loading commit data...
e2e_keypoint_rcnn_R_50_FPN_1x.yaml Loading commit data...
e2e_mask_rcnn_R_101_FPN_1x.yaml Loading commit data...
e2e_mask_rcnn_R_50_C4_1x.yaml Loading commit data...
e2e_mask_rcnn_R_50_FPN_1x.yaml Loading commit data...
e2e_mask_rcnn_X_101_32x8d_FPN_1x.yaml Loading commit data...
e2e_mask_rcnn_fbnet.yaml Loading commit data...
e2e_mask_rcnn_fbnet_xirb16d_dsmask.yaml Loading commit data...
rpn_R_101_FPN_1x.yaml Loading commit data...
rpn_R_50_C4_1x.yaml Loading commit data...
rpn_R_50_FPN_1x.yaml Loading commit data...
rpn_X_101_32x8d_FPN_1x.yaml Loading commit data...