Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dreamer-V2 #43

Merged
merged 50 commits into from
Jul 4, 2023
Merged

Dreamer-V2 #43

merged 50 commits into from
Jul 4, 2023

Conversation

belerico
Copy link
Member

@belerico belerico commented Jun 25, 2023

Summary

This PR introduce the Dreamer-V2 algorithm from https://arxiv.org/abs/2010.02193

Type of Change

Please select the one relevant option below:

  • New feature (non-breaking change that adds functionality)

Checklist

Please confirm that the following tasks have been completed:

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

Screenshots or Visuals (Optional)

If applicable, please provide screenshots, diagrams, graphs, or videos of the changes, features or the error.

Additional Information (Optional)

Those are the training loss and rewards con DMC Walker-Walk with the standard hyperparameters found in dreamer_v2/args.py

image
image

Thank you for your contribution! Once you have filled out this template, please ensure that you have assigned the appropriate reviewers and that all tests have passed.

@belerico belerico changed the title WIP: Dreamer-V2 Dreamer-V2 Jul 2, 2023
@belerico belerico marked this pull request as ready for review July 2, 2023 08:56
@belerico
Copy link
Member Author

belerico commented Jul 2, 2023

Maybe it's better to account for the action repeat as we have done in Dreamer-V1 @michele-milesi

@belerico belerico requested a review from DavideTr8 July 3, 2023 09:27
@belerico belerico merged commit 3da0d11 into main Jul 4, 2023
@belerico belerico deleted the feature/dreamer-v2 branch July 5, 2023 14:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants