Add support for complex distributions

will-maclean / sb3-burn

Implementation of stable-baselines3 in rust with burn

MIT License

11 stars 0 forks source link

Add support for complex distributions #32

Open will-maclean opened 3 weeks ago

will-maclean commented 3 weeks ago

As outlined in this issue, support for complex distributions is limited in burn. Implement some functionality to allow for this support in the meantime. This will be required for #7

will-maclean commented 2 weeks ago

https://github.com/DLR-RM/stable-baselines3/blob/master/stable_baselines3/common/distributions.py

will-maclean commented 2 weeks ago

Blocks #37

will-maclean commented 1 week ago

Simple Distributions

[x] (trait)
[ ] Uniform
[x] Normal
[ ] Categorical
[ ] Bernoulli

Action Distribution (https://github.com/DLR-RM/stable-baselines3/blob/master/stable_baselines3/common/distributions.py)

[x] (trait)
[x] DiagGaussianDistribution
[ ] SquashedDiagGaussianDistribution
[ ] CategoricalDistribution
[ ] MultiCategoricalDistribution
[ ] BernoulliDistribution
[ ] StateDependentNoiseDistribution
[ ] TanhBijector