Problem when loading 2chdeep pre-trained models

szagoruyko / cvpr15deepcompare

Code and models for "Learning to Compare Image Patches via Convolutional Neural Networks"

http://imagine.enpc.fr/~zagoruys/deepcompare.html

Other

468 stars 169 forks source link

Problem when loading 2chdeep pre-trained models #17

Closed abursuc closed 8 years ago

abursuc commented 8 years ago

Hi, I'm trying to load your already trained models using the 2chdeep setup. There seems to be a problem with the way you saved them. This is the error I get when doing torch.load():

unknown Torch class <inn.SpatialMaxPooling>

Could you please check the exported models?

Thanks a lot.

szagoruyko commented 8 years ago

do luarocks install inn and before loading the network require 'inn'. this is an old module that was integrated into nn.

abursuc commented 8 years ago

Thanks for the help. I should have explored a bit further the matter. I just wanted to use and off the shelf model quickly for a task. Nice work, by the way!