CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-22 04:40:15 +12:00
Code Issues Releases Wiki Activity

use settings.validation_crop

Browse source
This commit is contained in:
nagadomi 2015-06-26 20:12:51 +09:00
parent 0a37cefe25
commit fd9dadd7a4
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/settings.lua
View file

@ -21,7 +21,7 @@ cmd:option("-data_dir", "./data", 'data directory')
cmd:option("-test", "images/miku_small.png", 'test image file')
cmd:option("-model_dir", "./models", 'model directory')
cmd:option("-method", "scale", '(noise|scale|noise_scale)')
cmd:option("-noise_level", 1, '(1|2)')
cmd:option("-noise_level", 1, '(0|1|2)')
cmd:option("-color", 'rgb', '(y|rgb)')
cmd:option("-scale", 2.0, 'scale')
cmd:option("-learning_rate", 0.00025, 'learning rate for adam')
@ -64,7 +64,7 @@ settings.images = string.format("%s/images.t7", settings.data_dir)
settings.image_list = string.format("%s/image_list.txt", settings.data_dir)
settings.validation_ratio = 0.1
settings.validation_crops = 40
settings.validation_crops = 20
local srcnn = require './srcnn'
if (settings.method == "scale" or settings.method == "noise_scale") and settings.scale == 4 then

2
train.lua
View file

@ -109,7 +109,7 @@ local function train()
end
local best_score = 100000.0
print("# make validation-set")
local valid_xy = make_validation_set(valid_x, transformer, 20)
local valid_xy = make_validation_set(valid_x, transformer, settings.validation_crop)
valid_x = nil
collectgarbage()