fix: dv3-imagination notebook - Githubissues

Eclectic-Sheep / sheeprl

Distributed Reinforcement Learning accelerated by Lightning Fabric

https://eclecticsheep.ai

Apache License 2.0

322 stars 33 forks source link

fix: dv3-imagination notebook #290

Closed michele-milesi closed 6 months ago

michele-milesi commented 6 months ago

Summary

Describe the purpose of the pull request, including:

Fix dreamer v3 imagination notebook

Type of Change

Please select the one relevant option below:

Bug fix (non-breaking change that solves an issue)

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.
[x] I have updated the documentation, if applicable.
[x] I have installed pre-commit and run locally for my code changes.

Screenshots or Visuals (Optional)

I have tested on MsPacman, and the generated GIFs are reported below.

Imagination Steps:
Real Observations collected during environment interaction:

Note:

The actions used during imagination are the same sampled during the environment interaction, i.e., imagine_actions = False in the notebook.