Explorar o código

Fix training mode

nagadomi %!s(int64=9) %!d(string=hai) anos
pai
achega
634046d5f0
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  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)