Update train.py

wuyanxs99 · Nov 17, 2018 · 81fc48b · 81fc48b
1 parent 5a905de
commit 81fc48b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -107,7 +107,7 @@ def train(epoch):
     epoch_loss = 0
     for iteration, batch in enumerate(trainDataloader, 0):
         varIn, varTar = Variable(batch[0]), Variable(batch[1])
-        varIn, varTar = varTar.float(), varIn.float()
+        varIn, varTar = varIn.float(), varTar.float()
 
         if args.cuda:
             varIn = varIn.cuda()