Skip to content

Conversation

@vmoens
Copy link
Collaborator

@vmoens vmoens commented Jan 10, 2023

Description

Fixes Habitat integration and doc.
The goal is to make a benchmark for Habitat as we do for dm_control and Atari.

cc @vincentpierre

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jan 10, 2023
@vmoens vmoens added the bug Something isn't working label Jan 10, 2023
@vmoens
Copy link
Collaborator Author

vmoens commented Jan 11, 2023

Added support for device casting.
You can now call env = HabitatEnv(..., from_pixels=True, device=3) and the whole rendering will be executed on the appropriate device. You can also call env.to(device), and the env will be restarted on device.

@vmoens vmoens merged commit a6316c4 into main Jan 11, 2023
@vmoens vmoens deleted the fix_habitat branch March 31, 2023 16:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants