3 Commits

Author	SHA1	Message	Date
AdilZouitine	ff82367c62	Refactor SAC policy with performance optimizations and multi-camera support ... - Introduced Ensemble and CriticHead classes for more efficient critic network handling - Added support for multiple camera inputs in observation encoder - Optimized image encoding by batching image processing - Updated configuration for ManiSkill environment with reduced image size and action scaling - Compiled critic networks for improved performance - Simplified normalization and ensemble handling in critic networks Co-authored-by: michel-aractingi <michel.aractingi@gmail.com>	2025-02-20 17:14:27 +00:00
Michel Aractingi	ff47c0b0d3	- Fixed big issue in the loading of the policy parameters sent by the learner to the actor -- pass only the actor to the update_policy_parameters and remove strict=False ... - Fixed big issue in the normalization of the actions in the `forward` function of the critic -- remove the `torch.no_grad` decorator in `normalize.py` in the normalization function - Fixed performance issue to boost the optimization frequency by setting the storage device to be the same as the device of learning. Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>	2025-02-19 16:22:51 +00:00
AdilZouitine	2f3370e42f	Add maniskill support. ... Co-authored-by: Michel Aractingi <michel.aractingi@gmail.com>	2025-02-14 19:53:29 +00:00