So, I have two layers:
rnn = nn.LSTM(din, dhid, bidirectional=True, batch_first=True)
hidden2tag = nn.Linear(2 * dhid, target_size)
How can I tie the weights of linear layer and LSTM?
So, I have two layers:
rnn = nn.LSTM(din, dhid, bidirectional=True, batch_first=True)
hidden2tag = nn.Linear(2 * dhid, target_size)
How can I tie the weights of linear layer and LSTM?