Fix gpu option
This commit is contained in:
parent
202453d6c5
commit
6ba6cfe1ff
|
@ -276,6 +276,7 @@ local function waifu2x()
|
|||
if opt.thread > 0 then
|
||||
torch.setnumthreads(opt.thread)
|
||||
end
|
||||
cutorch.setDevice(opt.gpu)
|
||||
if cudnn then
|
||||
cudnn.fastest = true
|
||||
if opt.l:len() > 0 then
|
||||
|
@ -293,6 +294,5 @@ local function waifu2x()
|
|||
else
|
||||
convert_frames(opt)
|
||||
end
|
||||
cutorch.setDevice(opt.gpu)
|
||||
end
|
||||
waifu2x()
|
||||
|
|
Loading…
Reference in a new issue