Skip to content

Commit

Permalink
Update _process_sequence() docstring (#219)
Browse files Browse the repository at this point in the history
Small modification in the documentation of `_process_sequence` to reflect the correct order (`hidden_state`, `cell_state`) of `lstm_states`.

See order here: https://pytorch.org/docs/stable/generated/torch.nn.LSTM.html
  • Loading branch information
rogerioagjr authored Dec 5, 2023
1 parent 707cb0f commit 9f333ff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sb3_contrib/common/recurrent/policies.py
Original file line number Diff line number Diff line change
Expand Up @@ -170,7 +170,7 @@ def _process_sequence(
Do a forward pass in the LSTM network.
:param features: Input tensor
:param lstm_states: previous cell and hidden states of the LSTM
:param lstm_states: previous hidden and cell states of the LSTM, respectively
:param episode_starts: Indicates when a new episode starts,
in that case, we need to reset LSTM states.
:param lstm: LSTM object.
Expand Down

0 comments on commit 9f333ff

Please sign in to comment.