Vincent Moens

Results 145 issues of 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)...

CLA Signed

bug
CLA Signed
Environments
Data

TODO: - [x] Doc

enhancement
CLA Signed

```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):...

enhancement
CLA Signed

bug
CLA Signed
Suitable for minor