1
0
Fork 0
mirror of synced 2024-06-01 10:39:30 +12:00

Set cuda():evaluate()

This commit is contained in:
nagadomi 2015-11-09 03:45:03 +09:00
parent 986ef34ffc
commit 4322b63750

View file

@ -17,6 +17,8 @@ local opt = cmd:parse(arg)
local model = torch.load(opt.model, opt.iformat)
if model then
w2nn.cleanup_model(model)
model:cuda()
model:evaluate()
torch.save(opt.model, model, opt.oformat)
else
error("model not found")