ConvLSTM_pytorch
ConvLSTM_pytorch copied to clipboard
[feature] add dropout
Add the dropout parameter to ConvLSTM, which defaults to 0.0. For num_layers to 1, dropout does not work; For num_layers greater than 1, dropout works on the output of each layer except the last.