search

megvii-research / BBN

The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition
https://arxiv.org/abs/1912.02413
MIT License
658 stars 101 forks source link

load model from pretrain model error #12

Open Vickeyhw opened 4 years ago

Vickeyhw commented 4 years ago

I encounter errors when I load bbn_res32_cifar model from pre-trained res50 model . It shows that " Unexpected keys in state_dict......". How can I solve it？

ZhouBoyan commented 4 years ago

For res32 model， I think you should load the res32 pretrained model, rather than res50. By the way, we usually train res32 from scratch.