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

split first

Browse source
This commit is contained in:
nagadomi 2016-10-08 17:22:16 +09:00
parent a604aa3b7a
commit 66bff68ef1
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
train.lua
View file

@ -382,6 +382,8 @@ local function plot(train, valid)
{'validation', torch.Tensor(valid), '-'}})
end
local function train()
local x = remove_small_image(torch.load(settings.images))
local train_x, valid_x = split_data(x, math.max(math.floor(settings.validation_rate * #x), 1))
local hist_train = {}
local hist_valid = {}
local model
@ -397,8 +399,6 @@ local function train()
local criterion = create_criterion(model)
local eval_metric = w2nn.ClippedMSECriterion(0, 1):cuda()
local x = remove_small_image(torch.load(settings.images))
local train_x, valid_x = split_data(x, math.max(math.floor(settings.validation_rate * #x), 1))
local adam_config = {
xLearningRate = settings.learning_rate,
xBatchSize = settings.batch_size,