├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── evaluation ├── README.md ├── dev.py ├── eval.sh ├── r3meval │ ├── __init__.py │ ├── core │ │ ├── config │ │ │ ├── BC_config.yaml │ │ │ ├── eval_config.yaml │ │ │ └── hydra │ │ │ │ ├── launcher │ │ │ │ └── local.yaml │ │ │ │ └── output │ │ │ │ └── local.yaml │ │ ├── eval.sh │ │ ├── eval_loop.py │ │ ├── hydra_eval_launcher.py │ │ ├── hydra_launcher.py │ │ ├── run.sh │ │ └── train_loop.py │ └── utils │ │ ├── __init__.py │ │ ├── behavior_cloning.py │ │ ├── fc_network.py │ │ ├── gaussian_mlp.py │ │ ├── gym_env.py │ │ ├── logger.py │ │ ├── obs_wrappers.py │ │ ├── sampling.py │ │ ├── tensor_utils.py │ │ └── visualizations.py ├── run.sh └── setup.py ├── plots └── mvp_dino_r3m.py ├── r3m ├── __init__.py ├── cfgs │ ├── config_rep.yaml │ └── hydra │ │ ├── launcher │ │ └── local.yaml │ │ └── output │ │ └── local.yaml ├── example.py ├── models │ ├── models_language.py │ └── models_r3m.py ├── r3m_base.yaml ├── train_representation.py ├── trainer.py └── utils │ ├── data_loaders.py │ ├── logger.py │ └── utils.py ├── setup.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/README.md -------------------------------------------------------------------------------- /evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/README.md -------------------------------------------------------------------------------- /evaluation/dev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/dev.py -------------------------------------------------------------------------------- /evaluation/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/eval.sh -------------------------------------------------------------------------------- /evaluation/r3meval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/__init__.py -------------------------------------------------------------------------------- /evaluation/r3meval/core/config/BC_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/config/BC_config.yaml -------------------------------------------------------------------------------- /evaluation/r3meval/core/config/eval_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/config/eval_config.yaml -------------------------------------------------------------------------------- /evaluation/r3meval/core/config/hydra/launcher/local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/config/hydra/launcher/local.yaml -------------------------------------------------------------------------------- /evaluation/r3meval/core/config/hydra/output/local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/config/hydra/output/local.yaml -------------------------------------------------------------------------------- /evaluation/r3meval/core/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/eval.sh -------------------------------------------------------------------------------- /evaluation/r3meval/core/eval_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/eval_loop.py -------------------------------------------------------------------------------- /evaluation/r3meval/core/hydra_eval_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/hydra_eval_launcher.py -------------------------------------------------------------------------------- /evaluation/r3meval/core/hydra_launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/hydra_launcher.py -------------------------------------------------------------------------------- /evaluation/r3meval/core/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/run.sh -------------------------------------------------------------------------------- /evaluation/r3meval/core/train_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/core/train_loop.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/__init__.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/behavior_cloning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/behavior_cloning.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/fc_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/fc_network.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/gaussian_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/gaussian_mlp.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/gym_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/gym_env.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/logger.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/obs_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/obs_wrappers.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/sampling.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/tensor_utils.py -------------------------------------------------------------------------------- /evaluation/r3meval/utils/visualizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/r3meval/utils/visualizations.py -------------------------------------------------------------------------------- /evaluation/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/run.sh -------------------------------------------------------------------------------- /evaluation/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/evaluation/setup.py -------------------------------------------------------------------------------- /plots/mvp_dino_r3m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/plots/mvp_dino_r3m.py -------------------------------------------------------------------------------- /r3m/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/__init__.py -------------------------------------------------------------------------------- /r3m/cfgs/config_rep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/cfgs/config_rep.yaml -------------------------------------------------------------------------------- /r3m/cfgs/hydra/launcher/local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/cfgs/hydra/launcher/local.yaml -------------------------------------------------------------------------------- /r3m/cfgs/hydra/output/local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/cfgs/hydra/output/local.yaml -------------------------------------------------------------------------------- /r3m/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/example.py -------------------------------------------------------------------------------- /r3m/models/models_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/models/models_language.py -------------------------------------------------------------------------------- /r3m/models/models_r3m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/models/models_r3m.py -------------------------------------------------------------------------------- /r3m/r3m_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/r3m_base.yaml -------------------------------------------------------------------------------- /r3m/train_representation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/train_representation.py -------------------------------------------------------------------------------- /r3m/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/trainer.py -------------------------------------------------------------------------------- /r3m/utils/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/utils/data_loaders.py -------------------------------------------------------------------------------- /r3m/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/utils/logger.py -------------------------------------------------------------------------------- /r3m/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/r3m/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/setup.py -------------------------------------------------------------------------------- /visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stanford-iris-lab/segmenting_feats/HEAD/visualize.py --------------------------------------------------------------------------------