search

vignywang / IDC

Exploring Intrinsic Discrimination and Consistency for Weakly Supervised Object Localization
Other
2 stars 1 forks source link

Size mismatch while loading 29.pth.tar for evaluation #1

Closed akhilgakhar closed 1 month ago

akhilgakhar commented 2 months ago

size mismatch for classifier_loc.0.weight: copying a param with shape torch.Size([200, 768, 3, 3]) from checkpoint, the shape in current model is torch.Size([200, 256, 3, 3]).

vignywang commented 2 months ago

Please replace the latest networks/vgg_CUB.py

akhilgakhar commented 2 months ago

Please replace the latest networks/vgg_CUB.py

Thanks, that works :)