Skip to content

Conversation

@lukasmolnar
Copy link
Contributor

Issue ticket number and link

Fixes # (issue)

Describe your changes

Please include a summary of the change, including why you did this, and the desired effect.

Instructions for reviewers

Indicate anything in particular that you would like a code-reviewer to pay particular attention to.
Indicate steps to actually test code, including CLI instructions if different than usual.
Point out the desired behavior, and not just the "check that this appears" (otherwise the code reviewer will be lazy and just verify what you've already verified).

Checklist before requesting a review

  • This is expected to break regression tests.
  • I have assigned a reviewer
  • I have added the PR to the project, and tagged with with priority
  • If it is a core feature, I have added tests.
  • I have set up pre-commit hooks with ruff, or run ruff format . manually

Copy link
Owner

@sheim sheim left a comment

Choose a reason for hiding this comment

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

Mostly just code style comments for now : )

return (
torch.ones(self.latent_dim, self.num_actions).to(self.log_std.device) * std
)
return torch.ones(self.latent_dim, 1).to(self.log_std.device) * std
Copy link
Contributor Author

Choose a reason for hiding this comment

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

I need to double check this. It could be a bug in stable-baselines, since how they had it it seems the number of parameters was not reduced (by default it has shape (latent_dim, num_actions)).

@lukasmolnar lukasmolnar changed the base branch from dev to SAC July 2, 2024 19:00
Base automatically changed from SAC to dev July 11, 2024 00:16
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