CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-19 04:12:19 +12:00
Code Issues Releases Wiki Activity

Add thread option to benchmark

Browse source
This commit is contained in:
nagadomi 2016-04-15 09:15:19 +09:00
parent d31fbe9bb1
commit 8e9e4433d3
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tools/benchmark.lua
View file

@ -33,6 +33,7 @@ cmd:option("-show_progress", 1, 'show progressbar')
cmd:option("-baseline_filter", "Catrom", 'baseline interpolation (Box|Lanczos|Catrom(Bicubic))')
cmd:option("-save_info", 0, 'save score and parameters to benchmark.txt')
cmd:option("-save_all", 0, 'group -save_info, -save_image and -save_baseline_image option')
cmd:option("-thread", -1, 'number of CPU threads')
local function to_bool(settings, name)
if settings[name] == 1 then
@ -59,6 +60,9 @@ else
to_bool(opt, "save_baseline_image")
end
to_bool(opt, "show_progress")
if opt.thread > 0 then
torch.setnumthreads(tonumber(opt.thread))
end
local function rgb2y_matlab(x)
local y = torch.Tensor(1, x:size(2), x:size(3)):zero()