@@ -309,6 +309,7 @@ local function train()
instance_loss = torch.Tensor(x:size(1)):zero()
for i = 1, settings.inner_epoch do
+ model:training()
local train_score, il = minibatch_adam(model, criterion, eval_metric, x, y, adam_config)
instance_loss:copy(il)
print(train_score)