Fix resume
This commit is contained in:
parent
bdafab9e10
commit
4f07b74cc5
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue