[BUG] torch.distribution is not compatible with torchFX or TorchScript
Open
@vmoens is already working on this.
Since Jan 16, 2023.
bug
- Dominant language
- Python
- Stars
- 3.6k
- Forks
- 487
- Avg merge
- 1d 1h
- Merged PRs (30d)
- 207
Description
Sampling distribution is essential to RL methods.
Currerntly, torch.distributions cannot be compiled with TorchFX and TorchScript:
https://discuss.pytorch.org/t/how-to-serialize-in-torchscript-beta-distribution/168044
https://github.com/pytorch/pytorch/issues/29843
Some recipes can be found :
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.