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

Fix resume

Browse source
This commit is contained in:
nagadomi 2018-10-29 17:48:50 +00:00
parent bdafab9e10
commit 4f07b74cc5
1 changed files with 0 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
train.lua
View file

@ -556,12 +556,6 @@ local function train()
else
eval_metric = w2nn.ClippedMSECriterion():cuda()
end
local adam_config = {
xLearningRate = settings.learning_rate,
xBatchSize = settings.batch_size,
xLearningRateDecay = settings.learning_rate_decay,
xInstanceLoss = (settings.oracle_rate > 0)
}
local ch = nil
if settings.color == "y" then
ch = 1