-
### 🐛 Describe the bug
This error only happens when tensordict is imported (not used at all). Basically any backward pass after using flexattention fails.
```python
import torch
from einops im…
-
## Motivation
There are a range of cases where we would like to recover the original data.
## Solution
Adding a method to `VecNorm` to denormalize a tensordict or return a separate class for …
-
Hi guys,
I was tying to ICGNN within causica as follows:
I have executed:
!pip install causica
that loads fine.
but when I try to import ICGNN as below it fails.
from causica.functional_…
-
## Motivation
It is not very clear what should be the structure of TensorDict of the return of the _step() function for a multi agent environment.
If there are two agents A and B and they are in…
-
## Motivation
We can easily save tensordicts using `torchsnapshot`. However, as `MemmapTensors` store a file on disk, it would be fairly easy to save a tensordict that contains only MemmapTensors (…
-
## Motivation
A common situation for dataset generation/processing involves writing many tensors to disk from many processes/nodes in parallel, and over a long duration. While shared storage is ass…
-
Now that we have the generic Tensor and TensorDict, we can remove the old ambiguous and limited num_outputs and num_inputs from the dataset and replace it by extern_data using TensorDict and Tensor to…
-
Hello. why this error happened when I want to run this script: python mappo_ippo.py --m env.scenario_name=balance
## Describe the bug
Traceback (most recent call last):
File "mappo_ippo.py", …
-
Here we raise an exception if multiple tensors with a common storage have different dtypes:
https://github.com/pytorch/pytorch/blob/d21f311af880c736b18b5a588583f6162e9abcfa/torch/serialization.py#L…
-
https://pytorch.org/tensordict/