algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-07 10:05:29 +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

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

@ -64,5 +64,6 @@ class DecoderRNN(nn.Module):
predicted = outputs.max(1)[1] predicted = outputs.max(1)[1]
sampled_ids.append(predicted) sampled_ids.append(predicted)
inputs = self.embed(predicted) inputs = self.embed(predicted)
inputs = inputs.unsqueeze(1) # (batch_size, 1, embed_size)
sampled_ids = torch.cat(sampled_ids, 1) # (batch_size, 20) sampled_ids = torch.cat(sampled_ids, 1) # (batch_size, 20)
return sampled_ids.squeeze() return sampled_ids.squeeze()