瀏覽代碼

Fix training mode

nagadomi 9 年之前
父節點
當前提交
634046d5f0
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      train.lua

+ 1 - 0
train.lua

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