[Question] Item called in loss modules forces synchronization?
Open
@vmoens is already working on this.
Since Jan 10, 2024.
bug
- Dominant language
- Python
- Stars
- 3.6k
- Forks
- 484
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 207
Description
Aren't operations like this blocking? causing performance issues due to GPU synchronization?
https://github.com/pytorch/rl/blob/5b3dd988ad7f0f91a2f61f77a8b3380eae3131fd/torchrl/objectives/ppo.py#L443
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.