Skip to content

Commit

Permalink
Disable cudnn by default for #127. Add warning.
Browse files Browse the repository at this point in the history
  • Loading branch information
Brandon Amos committed Jun 7, 2016
1 parent 21f1bcc commit bced2bb
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
5 changes: 5 additions & 0 deletions training/main.lua
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,11 @@ if opt.cuda then
cutorch.setDevice(1)
end

if opt.cudnn then
print("\nThere is a known issue with cudnn training! See:")
print(" https://github.com/cmusatyalab/openface/issues/127\n\n")
end

os.execute('mkdir -p ' .. opt.save)
torch.save(paths.concat(opt.save, 'opts.t7'), opt, 'ascii')
print('Saving everything to: ' .. opt.save)
Expand Down
2 changes: 1 addition & 1 deletion training/opts.lua
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ function M.parse(arg)
'Home of dataset. Split into "train" and "val" directories that separate images by class.')
cmd:option('-manualSeed', 2, 'Manually set RNG seed')
cmd:option('-cuda', true, 'Use cuda.')
cmd:option('-cudnn', true, 'Convert the model to cudnn.')
cmd:option('-cudnn', false, 'Convert the model to cudnn.')

------------- Data options ------------------------
cmd:option('-nDonkeys', 2, 'number of donkeys to initialize (data loading threads)')
Expand Down

0 comments on commit bced2bb

Please sign in to comment.