Command Line Interface backwards compatible fix for models.py (#85)

* commandline backwards compatible fix for models.py * changes formatting to accomodate a 120 char width

Author: cuevasclemente

word_language_model/model.py

@@ -7,7 +7,15 @@ class RNNModel(nn.Module):
  def __init__(self, rnn_type, ntoken, ninp, nhid, nlayers):
  super(RNNModel, self).__init__()
  self.encoder = nn.Embedding(ntoken, ninp)
- self.rnn = getattr(nn, rnn_type)(ninp, nhid, nlayers, bias=False)
+ if rnn_type in ['LSTM', 'GRU']:
+ self.rnn = getattr(nn, rnn_type)(ninp, nhid, nlayers, bias=False)
+ else:
+ try:
+ nonlinearity = {'RNN_TANH': 'tanh', 'RNN_RELU': 'relu'}[rnn_type]
+ except KeyError:
+ raise ValueError( """An invalid option for `--model` was supplied,
+ options are ['LSTM', 'GRU', 'RNN_TANH' or 'RNN_RELU']""")
+ self.rnn = nn.RNN(ninp, nhid, nlayers, nonlinearity=nonlinearity, bias=False)
  self.decoder = nn.Linear(nhid, ntoken)
 
  self.init_weights()