CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-18 11:52:17 +12:00
Code Issues Releases Wiki Activity

Change default -validation_crops to 200.

Browse source
This commit is contained in:
nagadomi 2016-06-20 15:56:39 +09:00
parent d8fcde46f3
commit de669d24a4
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
train.lua
View file

@ -37,10 +37,11 @@ local function split_data(x, test_size)
end
local function make_validation_set(x, transformer, n, patches)
n = n or 4
local validation_patches = math.min(16, patches or 16)
local data = {}
for i = 1, #x do
for k = 1, math.max(n / patches, 1) do
local xy = transformer(x[i], true, patches)
for k = 1, math.max(n / validation_patches, 1) do
local xy = transformer(x[i], true, validation_patches)
for j = 1, #xy do
table.insert(data, {x = xy[j][1], y = xy[j][2]})
end