liuzhuang13 / DenseNet

Densely Connected Convolutional Networks, In CVPR 2017 (Best Paper Award).
BSD 3-Clause "New" or "Revised" License
4.71k stars 1.07k forks source link

error when loading pretrained model?? #12

Closed eriche2016 closed 7 years ago

eriche2016 commented 7 years ago

When i load the model with 201 layers which are pretrained on ImageNet, it output error message below:

torch/install/share/lua/5.1/nn/SpatialMaxPooling.lua:47: attempt to index field 'THNN' (a nil value)

it seems that the version of the pretrained model is not compatible with the latest torch packages. By the way, i am using the latest version of torch packages. can you provide a pretrained model using the lastest version of torch?

eriche2016 commented 7 years ago

my mistake, i forgot to load cunn package.