├── .github └── workflows │ └── codeql-analysis.yml ├── .gitignore ├── Atari ├── README.md ├── requirements.txt ├── scripts │ ├── __init__.py │ └── run.py └── src │ ├── __init__.py │ ├── agent.py │ ├── algos.py │ ├── masking_generator.py │ ├── models.py │ ├── rlpyt_atari_env.py │ ├── rlpyt_buffer.py │ ├── rlpyt_utils.py │ ├── utils.py │ └── vit_modules.py ├── CODE_OF_CONDUCT.md ├── DMControl ├── README.md ├── requirements.txt └── src │ ├── base_sac.py │ ├── config.yaml │ ├── configs │ ├── ball_in_cup_catch.yaml │ ├── cartpole_swingup.yaml │ ├── cheetah_run.yaml │ ├── finger_spin.yaml │ ├── reacher_easy.yaml │ └── walker_walk.yaml │ ├── curl_sac.py │ ├── cycdm_sac.py │ ├── decoder.py │ ├── encoder.py │ ├── idm_sac.py │ ├── logger.py │ ├── masking_generator.py │ ├── mtm_sac.py │ ├── spr_sac.py │ ├── train.py │ ├── transition_model.py │ ├── utils.py │ ├── video.py │ ├── vit_modules.py │ └── vit_sac.py ├── LICENSE ├── NOTICE.md ├── README.md ├── SECURITY.md ├── SUPPORT.md └── figs ├── Atari_Distribution.png ├── Atari_Full.png ├── Atari_IQM_OG.png ├── DMC.png └── framework.png /.github/workflows/codeql-analysis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/.github/workflows/codeql-analysis.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/.gitignore -------------------------------------------------------------------------------- /Atari/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/README.md -------------------------------------------------------------------------------- /Atari/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/requirements.txt -------------------------------------------------------------------------------- /Atari/scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/scripts/__init__.py -------------------------------------------------------------------------------- /Atari/scripts/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/scripts/run.py -------------------------------------------------------------------------------- /Atari/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/__init__.py -------------------------------------------------------------------------------- /Atari/src/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/agent.py -------------------------------------------------------------------------------- /Atari/src/algos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/algos.py -------------------------------------------------------------------------------- /Atari/src/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/masking_generator.py -------------------------------------------------------------------------------- /Atari/src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/models.py -------------------------------------------------------------------------------- /Atari/src/rlpyt_atari_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/rlpyt_atari_env.py -------------------------------------------------------------------------------- /Atari/src/rlpyt_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/rlpyt_buffer.py -------------------------------------------------------------------------------- /Atari/src/rlpyt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/rlpyt_utils.py -------------------------------------------------------------------------------- /Atari/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/utils.py -------------------------------------------------------------------------------- /Atari/src/vit_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/Atari/src/vit_modules.py -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /DMControl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/README.md -------------------------------------------------------------------------------- /DMControl/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/requirements.txt -------------------------------------------------------------------------------- /DMControl/src/base_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/base_sac.py -------------------------------------------------------------------------------- /DMControl/src/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/config.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/ball_in_cup_catch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/ball_in_cup_catch.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/cartpole_swingup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/cartpole_swingup.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/cheetah_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/cheetah_run.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/finger_spin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/finger_spin.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/reacher_easy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/reacher_easy.yaml -------------------------------------------------------------------------------- /DMControl/src/configs/walker_walk.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/configs/walker_walk.yaml -------------------------------------------------------------------------------- /DMControl/src/curl_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/curl_sac.py -------------------------------------------------------------------------------- /DMControl/src/cycdm_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/cycdm_sac.py -------------------------------------------------------------------------------- /DMControl/src/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/decoder.py -------------------------------------------------------------------------------- /DMControl/src/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/encoder.py -------------------------------------------------------------------------------- /DMControl/src/idm_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/idm_sac.py -------------------------------------------------------------------------------- /DMControl/src/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/logger.py -------------------------------------------------------------------------------- /DMControl/src/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/masking_generator.py -------------------------------------------------------------------------------- /DMControl/src/mtm_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/mtm_sac.py -------------------------------------------------------------------------------- /DMControl/src/spr_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/spr_sac.py -------------------------------------------------------------------------------- /DMControl/src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/train.py -------------------------------------------------------------------------------- /DMControl/src/transition_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/transition_model.py -------------------------------------------------------------------------------- /DMControl/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/utils.py -------------------------------------------------------------------------------- /DMControl/src/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/video.py -------------------------------------------------------------------------------- /DMControl/src/vit_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/vit_modules.py -------------------------------------------------------------------------------- /DMControl/src/vit_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/DMControl/src/vit_sac.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/NOTICE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/SECURITY.md -------------------------------------------------------------------------------- /SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/SUPPORT.md -------------------------------------------------------------------------------- /figs/Atari_Distribution.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/figs/Atari_Distribution.png -------------------------------------------------------------------------------- /figs/Atari_Full.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/figs/Atari_Full.png -------------------------------------------------------------------------------- /figs/Atari_IQM_OG.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/figs/Atari_IQM_OG.png -------------------------------------------------------------------------------- /figs/DMC.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/figs/DMC.png -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/Mask-based-Latent-Reconstruction/HEAD/figs/framework.png --------------------------------------------------------------------------------