Elad Sharony

icon location Ph.D. Student - Electrical & Computer Engineering - Technion | (RL)² Lab

Results 1 issues of Elad Sharony

[BUG] info['_weight'] device for Importance Sampling in PER

3 comment

## Describe the bug The device of `info['_weight']` doesn't match the storage device. ## To Reproduce ```python # From documentation from torchrl.data.replay_buffers import ReplayBuffer, LazyTensorStorage, PrioritizedSampler from tensordict import TensorDict...

bug