Ben Eisner

Resolves #204 . There are two callsites for rgb_handles_to_mask, one which reads directly from the sensor (and returns an array on the range [0, 1]) and one which reads from...

If you create demos with rendering say, at 256x256 resolution, and then load them using https://github.com/stepjam/RLBench/blob/master/rlbench/utils.py#L41 with CameraConfigs with different sizes (i.e. 128x128), then all images (including RGB + Depth)...

I've been using the dataset_generator.py script to generate demos for RLBench tasks, which is great. However, when I try to load them, I find that the segmentation mask handles are...

Support for ragged arrays, like torch.nested

1

One of the quite compelling new additions to the torch ecosystem is NestedTensors, see https://pytorch.org/docs/stable/nested.html. Basically, this is a new primitive in the Torch ecosystem which allows for tensors which...