Feature/diambra - Githubissues

Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric

https://eclecticsheep.ai

Apache License 2.0

300 stars 29 forks source link

Feature/diambra #65

Closed michele-milesi closed 1 year ago

michele-milesi commented 1 year ago

Summary

Describe the purpose of the pull request, including:

Added DiambraWrapper
Added Multi-encoder/decoder to all the algorithms, but Droq, Sac and PPO Recurrent
Added Multi-discrete support to PPO, DreamerV1 and P2E-DV1
Modified the make_env function to be able to train the agents on environments that return both pixel-like and vector-like observations
Modified the ReplayBuffer class to handle multiple observations

Type of Change

Please select the one relevant option below:

Breaking change (fix or feature that would cause existing functionality to not work as expected)

Checklist

Please confirm that the following tasks have been completed:

[x] I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
[x] I have added unit tests for my changes, or updated existing tests if necessary.
[ ] I have updated the documentation, if applicable.
[x] I have installed pre-commit and run locally for my code changes.