Skip to content

Match PPG implementation#186

Merged
vwxyzjn merged 32 commits intovwxyzjn:masterfrom
dipamc:ppg-dev
May 28, 2022
Merged

Match PPG implementation#186
vwxyzjn merged 32 commits intovwxyzjn:masterfrom
dipamc:ppg-dev

Conversation

@dipamc
Copy link
Copy Markdown
Collaborator

@dipamc dipamc commented May 18, 2022

Description

Types of changes

  • Bug fix
  • New feature
  • New algorithm
  • Documentation

Checklist:

  • I've read the CONTRIBUTION guide (required).
  • I have ensured pre-commit run --all-files passes (required).
  • I have updated the documentation and previewed the changes via mkdocs serve.
  • I have updated the tests accordingly (if applicable).

If you are adding new algorithms or your change could result in performance difference, you may need to (re-)run tracked experiments. See #137 as an example PR.

  • I have contacted @vwxyzjn to obtain access to the openrlbenchmark W&B team (required).
  • I have tracked applicable experiments in openrlbenchmark/cleanrl with --capture-video flag toggled on (required).
  • I have added additional documentation and previewed the changes via mkdocs serve.
    • I have explained note-worthy implementation details.
    • I have explained the logged metrics.
    • I have added links to the original paper and related papers (if applicable).
    • I have added links to the PR related to the algorithm.
    • I have created a table comparing my results against those from reputable sources (i.e., the original paper or other reference implementation).
    • I have added the learning curves (in PNG format with width=500 and height=300).
    • I have added links to the tracked experiments.
  • I have updated the tests accordingly (if applicable).

@vercel
Copy link
Copy Markdown

vercel bot commented May 18, 2022

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Updated
cleanrl ✅ Ready (Inspect) Visit Preview May 28, 2022 at 0:09AM (UTC)

@gitpod-io
Copy link
Copy Markdown

gitpod-io bot commented May 18, 2022

Copy link
Copy Markdown
Owner

@vwxyzjn vwxyzjn left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is massive! Thanks for making this PR.

I think before any additional changes, our next step is to establish a great baseline. We should try to compare our results against the original results, otherwise, it's hard to attest the quality of our PPG implementation given that we included divergent changes such as using shared optimizers.

The original paper ran experiments using the hard distribution mode, so we might have to re-run them. I made a fork available here https://github.com/openrlbenchmark/phasic-policy-gradient to run tracked experiments. Unfortunately, I was not able to run them due to insufficient GPU memory... Would you mind giving it a try? The benchmark commands is at https://github.com/openrlbenchmark/phasic-policy-gradient/blob/add-wandb/benchmark.sh

* Original PPO used orthogonal initialization of only the Policy head and Value heads with scale of 0.01 and 1. respectively.
* For PPG
* All weights are initialized with the default torch initialization (Kaiming Uniform)
* Each layer’s weights are divided by the L2 norm of the weights along the (which axis?), and multiplied by a scale factor.
Copy link
Copy Markdown
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please clarify "which axis" here.

@vwxyzjn vwxyzjn merged commit eba6452 into vwxyzjn:master May 28, 2022
ludgerpaehler pushed a commit to ludgerpaehler/koopman-rl that referenced this pull request Jan 13, 2026
* added nit changes from ppg code

* change observation buffer to uint8

* sample full rollouts

* minor device fix

* update optimizer settings

* add ppg documentation

* update mkdocs

* update images to png for codespell errors

* trigger CI

* Minor format change

* format by running `pre-commit`

* removes trailing space

* Add an extra note

* argument names and documentation changes

* add capture video

* add experiment report

* Update documentation

* Quick css fix

* Update documentation

* Fix documentation for PPO

* Add benchmark commands

* Add benchmark commands

* add metrics section

* Add more docs

* Quick fix on ddpg docs

* Add procgen test cases

* Update CI

* test CI

* test ci

* Update tests

* normalization axis documentation

Co-authored-by: Dipam Chakraborty <dipam@aicrowd.com>
Co-authored-by: Costa Huang <costa.huang@outlook.com>
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.

2 participants