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

CLEAN_MODEL=true by default. No overhead

Browse source
This commit is contained in:
nagadomi 2018-11-13 10:45:11 +09:00
parent 996ff1d3dc
commit 1b946a5484
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
web.lua
View file

@ -76,7 +76,7 @@ local photo_model = {
noise3 = w2nn.load_model(path.join(PHOTO_MODEL_DIR, "noise3_model.t7"), opt.force_cudnn) noise3 = w2nn.load_model(path.join(PHOTO_MODEL_DIR, "noise3_model.t7"), opt.force_cudnn)
} }
collectgarbage() collectgarbage()
local CLEANUP_MODEL = false -- if you are using the low memory GPU, you could use this flag. local CLEANUP_MODEL = true -- if you are using the low memory GPU, you could use this flag.
local CACHE_DIR = path.join(ROOT, "cache") local CACHE_DIR = path.join(ROOT, "cache")
local MAX_NOISE_IMAGE = opt.max_pixels local MAX_NOISE_IMAGE = opt.max_pixels
local MAX_SCALE_IMAGE = (math.sqrt(opt.max_pixels) / 2)^2 local MAX_SCALE_IMAGE = (math.sqrt(opt.max_pixels) / 2)^2