Commit f40f2993 authored by Preston Parry's avatar Preston Parry Committed by Francisco Massa

Fixes bug with double (overwriting) RESNETS params (#488)

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.
parent 539b1322
...@@ -5,6 +5,9 @@ MODEL: ...@@ -5,6 +5,9 @@ MODEL:
CONV_BODY: "R-152-FPN" CONV_BODY: "R-152-FPN"
RESNETS: RESNETS:
BACKBONE_OUT_CHANNELS: 256 BACKBONE_OUT_CHANNELS: 256
STRIDE_IN_1X1: False
NUM_GROUPS: 32
WIDTH_PER_GROUP: 8
RPN: RPN:
USE_FPN: True USE_FPN: True
ANCHOR_STRIDE: (4, 8, 16, 32, 64) ANCHOR_STRIDE: (4, 8, 16, 32, 64)
...@@ -28,10 +31,6 @@ MODEL: ...@@ -28,10 +31,6 @@ MODEL:
POOLER_SAMPLING_RATIO: 2 POOLER_SAMPLING_RATIO: 2
RESOLUTION: 28 RESOLUTION: 28
SHARE_BOX_FEATURE_EXTRACTOR: False SHARE_BOX_FEATURE_EXTRACTOR: False
RESNETS:
STRIDE_IN_1X1: False
NUM_GROUPS: 32
WIDTH_PER_GROUP: 8
MASK_ON: True MASK_ON: True
DATASETS: DATASETS:
TEST: ("coco_2014_minival",) TEST: ("coco_2014_minival",)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment