New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Runtime error when loading DenseNet161 model #67

Open

Joooshua opened this issue Sep 16, 2019 · 1 comment

Joooshua commented Sep 16, 2019

There are key mismatches while loading the pretrained DenseNet161 Model.

I think it's because the DenseNet structure varies from the previous version to the present version?

I'm wnodering which version of torchvision you are using when training the DenseNet161 model.

Thank u !!!!!

HuaizhengZhang commented May 9, 2020

https://github.com/HuaizhengZhang/scene-recognition-pytorch1.x-tf2.x

I have upgraded the models by using pytorch1.4. You may try.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment