|
@@ -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
|