Skip to content

Commit

Permalink
add a note: please use only 1 gpu to run LSTM, pytorch/pytorch#21108
Browse files Browse the repository at this point in the history
  • Loading branch information
zjzzqs committed Apr 26, 2020
1 parent a460dc9 commit 221e233
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/yews/models/polarity.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,6 +212,8 @@ def polarity_v2(pretrained=False, progress=True, **kwargs):
model.load_state_dict(state_dict)
return model


# note: please use only 1 gpu to run LSTM, https://github.com/pytorch/pytorch/issues/21108
class PolarityLSTM(nn.Module):
r"""a LSTM neural network
@author: Chujie Chen
Expand Down

0 comments on commit 221e233

Please sign in to comment.