(1). My pytorch version is 1.4.0, and my data class is 1, when I train this model ,I encountered this problem:
RuntimeError: Error(s) in loading state_dict for FSSDLite:
size mismatch for conf.0.weight: copying a param with shape torch.Size([126, 512, 3, 3]) from checkpoint, the shape in current model is torch.Size([12, 512, 3, 3]).
......
size mismatch for conf.5.bias: copying a param with shape torch.Size([84]) from checkpoint, the shape in current model is torch.Size([8]).
I have change the num_classes to 2. Is it just because my pytoch version or other problems? Thank you for any reply!
(2) In FSSD-MobileNet, I want to change the number of fused feature maps, such as make four feature maps to fuse together, which part should I change?
(1). My pytorch version is 1.4.0, and my data class is 1, when I train this model ,I encountered this problem: RuntimeError: Error(s) in loading state_dict for FSSDLite: size mismatch for conf.0.weight: copying a param with shape torch.Size([126, 512, 3, 3]) from checkpoint, the shape in current model is torch.Size([12, 512, 3, 3]). ...... size mismatch for conf.5.bias: copying a param with shape torch.Size([84]) from checkpoint, the shape in current model is torch.Size([8]). I have change the num_classes to 2. Is it just because my pytoch version or other problems? Thank you for any reply!
(2) In FSSD-MobileNet, I want to change the number of fused feature maps, such as make four feature maps to fuse together, which part should I change?