wonderful codes
chenmozxh opened this issue · 1 comments
chenmozxh commented
wonderful codes!!!
line 214-218 in model.py may be modified to:
out = torch.unsqueeze(torch.squeeze(_), 1)
fartashf commented
Not sure how you can do so. Lines 214-218 take the hiddens of the RNN and extract the last hiddens for sequence. out is PackedSequence object and I is a tensor with indices of the last hiddens.