search

soumith / imagenet-multiGPU.torch

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

opts.backend not supported, breaks model.lua #56

Closed anuragranj closed 8 years ago

anuragranj commented 8 years ago

In opts.lua :

cmd:option('-backend',     'cudnn', 'Options: cudnn | ccn2 | cunn')

selecting any other option apart from cudnn breaks at, model.lua :

if opt.backend == 'cudnn' then
    require 'cudnn'
    cudnn.convert(model, cudnn)
 elseif opt.backend ~= 'nn' then
    error'Unsupported backend'

Since the above script only supports 'cudnn'. We should consider either updating opts.lua or model.lua.

szagoruyko commented 8 years ago

it supports nn and cudnn, but yes you're right we should change opts.lua

soumith commented 8 years ago

fixed help in master