Vincent Moens
Vincent Moens
This PR introduces the concept of storage save hooks, which can modify a storage before serializing it TODO: - Account for more complex data structures (e.g. nested done, nested rewards)...
TODO: - [x] Doc
```python from torchrl.modules import LSTM import torch from torch.utils.benchmark import Timer from torchrl.modules.tensordict_module.rnn import _get_num_per_traj_init, _split_and_pad_sequence b = 10 t = 100 c = 32 device = "cuda" with torch.device(device):...