fix for this code: "imgs = encoder(self.fake_imgs[0])", need to load in all imgs with a loop #5

Author: hughkf

code/trainer.py

@@ -473,8 +473,8 @@ def train(self):
                     encoder.fine_tune(fine_tune_encoder)
                     encoder_optimizer = torch.optim.Adam(params=filter(lambda p: p.requires_grad, encoder.parameters()),
                                                          lr=1e-4) if fine_tune_encoder else None
-                    imgs = encoder(self.fake_imgs[i].cuda())
-                    scores, caps_sorted, decode_lengths, alphas, sort_ind = decoder(imgs, caps, caplens)
+                    imgs = encoder(self.fake_imgs[i]).cuda()
+                    scores, caps_sorted, decode_lengths, alphas, sort_ind = decoder(imgs.cuda(), caps.cuda(), caplens.cuda()).cuda()
                     targets = caps_sorted[:, 1:]
                     scores, _ = pack_padded_sequence(scores, decode_lengths, batch_first=True).cuda()
                     targets, _ = pack_padded_sequence(targets, decode_lengths, batch_first=True).cuda()