Elad Sharony comments

Repositories
Issues
Comments

Results 1 comments of


                                            Elad Sharony

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

> Also, the sampler is unaware of what the storage is. You could have multiple storages for instance. Maybe I'm missing something, but [def sample(self, storage: Storage, batch_size: int)](https://github.com/pytorch/rl/blob/c851e1698dd4a46fd1429f5afe8057c80072475e/torchrl/data/replay_buffers/samplers.py#L484C9-L484C15) accepts...