problem regarding vgg model weights loading

marcellacornia / mlnet

A Deep Multi-Level Network for Saliency Prediction. ICPR 2016

MIT License

94 stars 37 forks source link

problem regarding vgg model weights loading #2

Closed shaymush closed 7 years ago

shaymush commented 7 years ago

Hi, I'm trying to run your code in test mode and I get an exception error: Layer weight shape(3,3,640,64) not compatible with provided weight shape (64,3,3,3) What could be the problem?

marcellacornia commented 7 years ago

Hi @shaymush, which backend are you using? MLNet supports Theano only.

shaymush commented 7 years ago

I'm using Theano backend...

marcellacornia commented 7 years ago

Have you set the dim ordering to "th" in the keras.json file?

shaymush commented 7 years ago

You're right! I forgot to change it. Now it works. Thanks!