search

soumith / imagenet-multiGPU.torch

an imagenet example in torch.
BSD 2-Clause "Simplified" License
401 stars 158 forks source link

Retrain throws error #44

Closed ppwwyyxx closed 8 years ago

ppwwyyxx commented 8 years ago

Use -retrain on a cudnn model throws error like cannot find cudnn.SpatialSoftmax. Adding this will fix the problem.

soumith commented 8 years ago

Thanks. cc: @szagoruyko fyi

ppwwyyxx commented 8 years ago

BTW, maybe there could be problems on cunn backend, too? Didn't test that.