├── .gitignore ├── LICENSE ├── README.md ├── bc ├── diffusion_unet.py └── diffusion_unet_rgbd.py ├── env_wrappers ├── better_obs.py └── better_rewards.py ├── environment.yml ├── nets ├── cnn │ └── plain_conv.py └── diffusion_policy │ └── conditional_unet1d_colab.py ├── rl ├── ppo_rgbd.py ├── ppo_state.py ├── sac_rgbd.py └── sac_state.py └── utils ├── ms2_data.py ├── profiling.py ├── sampler.py └── torch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/README.md -------------------------------------------------------------------------------- /bc/diffusion_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/bc/diffusion_unet.py -------------------------------------------------------------------------------- /bc/diffusion_unet_rgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/bc/diffusion_unet_rgbd.py -------------------------------------------------------------------------------- /env_wrappers/better_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/env_wrappers/better_obs.py -------------------------------------------------------------------------------- /env_wrappers/better_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/env_wrappers/better_rewards.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/environment.yml -------------------------------------------------------------------------------- /nets/cnn/plain_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/nets/cnn/plain_conv.py -------------------------------------------------------------------------------- /nets/diffusion_policy/conditional_unet1d_colab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/nets/diffusion_policy/conditional_unet1d_colab.py -------------------------------------------------------------------------------- /rl/ppo_rgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/rl/ppo_rgbd.py -------------------------------------------------------------------------------- /rl/ppo_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/rl/ppo_state.py -------------------------------------------------------------------------------- /rl/sac_rgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/rl/sac_rgbd.py -------------------------------------------------------------------------------- /rl/sac_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/rl/sac_state.py -------------------------------------------------------------------------------- /utils/ms2_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/utils/ms2_data.py -------------------------------------------------------------------------------- /utils/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/utils/profiling.py -------------------------------------------------------------------------------- /utils/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/utils/sampler.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tongzhoumu/ManiSkill_Baselines/HEAD/utils/torch_utils.py --------------------------------------------------------------------------------