-
Notifications
You must be signed in to change notification settings - Fork 505
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
Overhaul Observation Transformation API + Better 360 and multisensor support. #478
Conversation
Co-authored-by: Oleksandr <maksymets.o@gmail.com>
All tests are now passing. Ready to get this merged. Only thing I am unsure of is the ObsTransformer name and if I should move it from the RL.POLICY to just RL. |
I am going to move the OBS_TRANSFORMER out of RL.POLICY to RL. and willl add the doc strings. |
I am still not sure what scope to put the config under, thoughts @erikwijmans I could RL.OBS_TRANSFORMER, but I could also see the arguement for RL.POLICY.OBS_TRANSFORMER. It's general enough to just be at the root node too though. |
* [esp/sim] collapse SimulatorWithAgents into Simulator * Fix simulator reset behaviour to reset Agent to initial state.
…support. (facebookresearch#478) * Add UUID config to simulator visual sensors * Add comments about UUID and add to BumpSensor * Incorporate suggestions * Remove unused code * Make base sensor have overidable UUID * Added observation transformer config values * Remove duplicated class * Bugfixes and missing fromConfig params * Remove debug statements and add more type hints * More typehints * Improve docstring * Apply suggestions from code review Co-authored-by: Oleksandr <maksymets.o@gmail.com> * Address most of the comments * Refactor to use attr * Update API to have access to entire dict * bugfixes * More bugfixes * more bugfixes * Major refactor + Obs_Transformer overhaul * Fix license comment Co-authored-by: Oleksandr <maksymets.o@gmail.com> * Remove observation transformers * Fix errors and merge conflicts * Fix issues, refactor and add tests * Fix minor bug * address comments * Fix test and doc bug * More docstring improvements * Fix bug in ResizeShortestEdge * remove attrs from nn.module * Fix remaining bugs and refactor Cube2Equirec for smart device selection * Improve obs_space handling in PPO and change defaults of Cube2EQ * more doc strings * bump for ci
…support. (facebookresearch#478) * Add UUID config to simulator visual sensors * Add comments about UUID and add to BumpSensor * Incorporate suggestions * Remove unused code * Make base sensor have overidable UUID * Added observation transformer config values * Remove duplicated class * Bugfixes and missing fromConfig params * Remove debug statements and add more type hints * More typehints * Improve docstring * Apply suggestions from code review Co-authored-by: Oleksandr <maksymets.o@gmail.com> * Address most of the comments * Refactor to use attr * Update API to have access to entire dict * bugfixes * More bugfixes * more bugfixes * Major refactor + Obs_Transformer overhaul * Fix license comment Co-authored-by: Oleksandr <maksymets.o@gmail.com> * Remove observation transformers * Fix errors and merge conflicts * Fix issues, refactor and add tests * Fix minor bug * address comments * Fix test and doc bug * More docstring improvements * Fix bug in ResizeShortestEdge * remove attrs from nn.module * Fix remaining bugs and refactor Cube2Equirec for smart device selection * Improve obs_space handling in PPO and change defaults of Cube2EQ * more doc strings * bump for ci
Motivation and Context
Solves #367
Okay, this ObservationTransformer API does the following:
How Has This Been Tested
With Pytest and by splitting the ResizeCenterCropper into two PyTorch modules
Types of changes
Checklist