STDC-Seg icon indicating copy to clipboard operation
STDC-Seg copied to clipboard

Issues about loading pretrained model.

Open Windaway opened this issue 4 years ago • 0 comments

The implement uses init_weight to load the pretrained model. However, you use nn.Sequential to repackage the nn.Modules. So, the code has two sets of parmaters, and you only load weight to the first sets. That is, the second set (for segmentation) are trained from scratch.

Windaway avatar Jan 10 '22 04:01 Windaway