Skip to content

chore: remove unused parameters in jax implementations#264

Merged
vwxyzjn merged 1 commit intovwxyzjn:masterfrom
kinalmehta:jax-args-fix
Aug 22, 2022
Merged

chore: remove unused parameters in jax implementations#264
vwxyzjn merged 1 commit intovwxyzjn:masterfrom
kinalmehta:jax-args-fix

Conversation

@kinalmehta
Copy link
Copy Markdown
Collaborator

Description

This PR refactors the *_jax.py implementations as follows:

  • removes unused parameters from argparse (torch-deterministic as jax is deterministic by default, cuda as GPU is activated by default if available)
  • fix pybullet envs in td3_jax and ddpg_jax

Given this is a non-performance-impacting refactor, re-running the benchmark is not required.

fix pybullet envs in td3_jax and ddpg_jax
@vercel
Copy link
Copy Markdown

vercel bot commented Aug 22, 2022

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

Name Status Preview Updated
cleanrl ✅ Ready (Inspect) Visit Preview Aug 22, 2022 at 9:25PM (UTC)

@vwxyzjn
Copy link
Copy Markdown
Owner

vwxyzjn commented Aug 22, 2022

Thank you! @kinalmehta

@vwxyzjn vwxyzjn merged commit 5c461e1 into vwxyzjn:master Aug 22, 2022
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