Update model.py

Author: yunjey

tutorials/03-advanced/image_captioning/model.py

@@ -64,5 +64,6 @@ def sample(self, features, states=None):
             predicted = outputs.max(1)[1]
             sampled_ids.append(predicted)
             inputs = self.embed(predicted)
+            inputs = inputs.unsqueeze(1)                         # (batch_size, 1, embed_size)
         sampled_ids = torch.cat(sampled_ids, 1)                  # (batch_size, 20)
-        return sampled_ids.squeeze()
+        return sampled_ids.squeeze()