wcneill · July 30, 2020 03:40
diff --git a/lstm03.py b/lstm03.py
 input_size  = 50  # representing the one-hot encoded vector size
 hidden_size = 100 # number of hidden nodes in the LSTM layer
 n_layers    = 2   # number of LSTM layers
 output_size = 50  # output of 50 scores for the next character

 lstm   = nn.LSTM(input_size, n_hidden, n_layers, batch_first=True)
 linear = nn.Linear(n_hidden, output_size)
	input_size = 50 # representing the one-hot encoded vector size
	hidden_size = 100 # number of hidden nodes in the LSTM layer
	n_layers = 2 # number of LSTM layers
	output_size = 50 # output of 50 scores for the next character

	lstm = nn.LSTM(input_size, n_hidden, n_layers, batch_first=True)
	linear = nn.Linear(n_hidden, output_size)