yangze0930 / NTS-Net

This is a PyTorch implementation of the ECCV2018 paper "Learning to Navigate for Fine-grained Classification" (Ze Yang, Tiange Luo, Dong Wang, Zhiqiang Hu, Jun Gao, Liwei Wang).
MIT License
452 stars 118 forks source link

矩阵大小不一致 #13

Closed zhaozhiyong19890102 closed 5 years ago

zhaozhiyong19890102 commented 5 years ago

size mismatch for concat_net.weight: copying a param with shape torch.Size([200, 10240]) from checkpoint, the shape in current model is torch.Size([200, 6144])