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

Fix gpu option

Browse source
This commit is contained in:
nagadomi 2017-04-01 03:55:09 +09:00
parent 202453d6c5
commit 6ba6cfe1ff
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
waifu2x.lua
View file

@ -276,6 +276,7 @@ local function waifu2x()
if opt.thread > 0 then if opt.thread > 0 then
torch.setnumthreads(opt.thread) torch.setnumthreads(opt.thread)
end end
cutorch.setDevice(opt.gpu)
if cudnn then if cudnn then
cudnn.fastest = true cudnn.fastest = true
if opt.l:len() > 0 then if opt.l:len() > 0 then
@ -293,6 +294,5 @@ local function waifu2x()
else else
convert_frames(opt) convert_frames(opt)
end end
cutorch.setDevice(opt.gpu)
end end
waifu2x() waifu2x()