def __init__(self, batch_size, seq_length, vocabulary_size, lstm_num_hidden=256, lstm_num_layers=2, device='cuda:0', input_size=1):
    super(TextGenerationModel, self).__init__()
    self.emb_size = 64
    self.device = device
    self.lstm = nn.LSTM(input_size, lstm_num_hidden, num_layers=lstm_num_layers, dropout=0)
    self.linear = nn.Linear(lstm_num_hidden, vocabulary_size)
    self.h = None
def __init__(self, batch_size, seq_length, vocabulary_size, lstm_num_hidden=256, lstm_num_layers=2, device='cuda:0', input_size=1):
    super(TextGenerationModel, self).__init__()
    self.emb_size = 64
    self.device = device
    self.lstm = nn.LSTM(input_size, lstm_num_hidden, num_layers=lstm_num_layers, dropout=0)
    self.linear = nn.Linear(lstm_num_hidden, vocabulary_size)
    self.h = None