[Feature Request] Implement Efficient Ops to Compute Returns
Open
@vmoens is already working on this.
Since Aug 29, 2022.
enhancement
- Dominant language
- Python
- Stars
- 3.6k
- Forks
- 484
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 207
Description
In RL algorithms, it is every common to compute return-like things from trajectories. It is inefficient to compute such returns with normal python for-loops. In order to improve efficiency, we'd better to abstract some general return ops and implement efficient PyTorch kernels. Some examples of such returns are listed below.
- TD-lambda return
- GAE in PPO
- ...
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.