Pārlūkot izejas kodu

Fix training mode

nagadomi 9 gadi atpakaļ
vecāks
revīzija
634046d5f0
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  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)