search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

Why set: 'upsample.upconv.weight': None in keypoint net? #191

Closed lilichu closed 5 years ago

lilichu commented 5 years ago

Why set 'upsample.upconv.weight': None? In this way we can not load upsample.upconv.weight when load ckpt. but why? thanks!

https://github.com/roytseng-tw/Detectron.pytorch/blob/8315af319cd29b8884a7c0382c4700a96bf35bbc/lib/modeling/keypoint_rcnn_heads.py#L70

        if self.upsample_heatmap:
            blob_name = 'kps_score_lowres'
            detectron_weight_mapping.update({
                'upsample.upconv.weight': None,  # 0: don't load from or save to checkpoint
                'upsample.upconv.bias': None
})