Stable-Baselines-Team / stable-baselines3-contrib

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code
https://sb3-contrib.readthedocs.io
MIT License
465 stars 173 forks source link

Update `_process_sequence()` docstring #219

Closed rogerioagjr closed 10 months ago

rogerioagjr commented 10 months ago

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

Description

Context

Types of changes

Checklist:

Note: we are using a maximum length of 127 characters per line