algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-06 17:44:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
yunjey
2017-09-28 15:33:17 +09:00
committed by GitHub
parent 17ef894aec
commit 1ec90fad72
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tutorials/03-advanced/image_captioning/model.py
View File

@ -64,5 +64,6 @@ class DecoderRNN(nn.Module):
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()