Open mobeets opened 1 year ago
use_lstm is actually using LSTMWrapper, which happens here.
If there's a way to do this manually, that would be cool. basically, the goal would be to make an r2d2 model that simply uses GRU instead of LSTM
use_lstm is actually using LSTMWrapper, which happens here.
If there's a way to do this manually, that would be cool. basically, the goal would be to make an r2d2 model that simply uses GRU instead of LSTM