rearrange code for cuda()

yunjey · yunjey · commit 2fe796bb1000 · 2017-03-22T22:58:29.000+09:00
diff --git a/tutorials/09 - Image Captioning/train.py b/tutorials/09 - Image Captioning/train.py
@@ -55,12 +55,11 @@ def main():
             # Set mini-batch dataset
             images = Variable(images)
             captions = Variable(captions)
-            targets = pack_padded_sequence(captions, lengths, batch_first=True)[0]
-            
             if torch.cuda.is_available():
                 images = images.cuda()
                 captions = captions.cuda()
-
+            targets = pack_padded_sequence(captions, lengths, batch_first=True)[0]
+            
             # Forward, Backward and Optimize
             decoder.zero_grad()
             encoder.zero_grad()
