CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-14 08:54:31 +12:00
Code Issues Releases Wiki Activity

Add -backend option to use cuDNN

Browse source
This commit is contained in:
nagadomi 2015-10-31 04:38:28 +09:00
parent 9ddee6088d
commit 243d8821be
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/settings.lua
View file

@ -19,7 +19,7 @@ cmd:text("waifu2x-training")
cmd:text("Options:")
cmd:option("-seed", 11, 'fixed input seed')
cmd:option("-data_dir", "./data", 'data directory')
-- cmd:option("-backend", "cunn", '(cunn|cudnn)') -- cudnn is slower than cunn
cmd:option("-backend", "cunn", '(cunn|cudnn)')
cmd:option("-test", "images/miku_small.png", 'test image file')
cmd:option("-model_dir", "./models", 'model directory')
cmd:option("-method", "scale", '(noise|scale)')
@ -91,6 +91,4 @@ end
settings.images = string.format("%s/images.t7", settings.data_dir)
settings.image_list = string.format("%s/image_list.txt", settings.data_dir)
settings.backend = "cunn"
return settings