├── .gitignore ├── LICENSE.md ├── README.md ├── data4robotics ├── __init__.py ├── agent.py ├── load_pretrained.py ├── misc.py ├── models │ ├── __init__.py │ ├── action_distributions.py │ ├── base.py │ ├── resnet.py │ └── vit.py ├── replay_buffer.py ├── task.py ├── trainers │ ├── __init__.py │ ├── base.py │ └── bc.py └── transforms.py ├── download_features.sh ├── env.yml ├── experiments ├── agent │ ├── default.yaml │ ├── features │ │ ├── r3m.yaml │ │ ├── resnet_gn.yaml │ │ └── vit_base.yaml │ └── policy │ │ ├── gaussian_constant.yaml │ │ └── gaussian_mixture.yaml ├── finetune.yaml ├── hydra │ └── launcher │ │ └── slurm.yaml ├── task │ └── franka.yaml └── trainer │ └── bc.yaml ├── finetune.py ├── pretrained_networks_example.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/README.md -------------------------------------------------------------------------------- /data4robotics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/__init__.py -------------------------------------------------------------------------------- /data4robotics/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/agent.py -------------------------------------------------------------------------------- /data4robotics/load_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/load_pretrained.py -------------------------------------------------------------------------------- /data4robotics/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/misc.py -------------------------------------------------------------------------------- /data4robotics/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/models/__init__.py -------------------------------------------------------------------------------- /data4robotics/models/action_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/models/action_distributions.py -------------------------------------------------------------------------------- /data4robotics/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/models/base.py -------------------------------------------------------------------------------- /data4robotics/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/models/resnet.py -------------------------------------------------------------------------------- /data4robotics/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/models/vit.py -------------------------------------------------------------------------------- /data4robotics/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/replay_buffer.py -------------------------------------------------------------------------------- /data4robotics/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/task.py -------------------------------------------------------------------------------- /data4robotics/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/trainers/__init__.py -------------------------------------------------------------------------------- /data4robotics/trainers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/trainers/base.py -------------------------------------------------------------------------------- /data4robotics/trainers/bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/trainers/bc.py -------------------------------------------------------------------------------- /data4robotics/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/data4robotics/transforms.py -------------------------------------------------------------------------------- /download_features.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/download_features.sh -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/env.yml -------------------------------------------------------------------------------- /experiments/agent/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/default.yaml -------------------------------------------------------------------------------- /experiments/agent/features/r3m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/features/r3m.yaml -------------------------------------------------------------------------------- /experiments/agent/features/resnet_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/features/resnet_gn.yaml -------------------------------------------------------------------------------- /experiments/agent/features/vit_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/features/vit_base.yaml -------------------------------------------------------------------------------- /experiments/agent/policy/gaussian_constant.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/policy/gaussian_constant.yaml -------------------------------------------------------------------------------- /experiments/agent/policy/gaussian_mixture.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/agent/policy/gaussian_mixture.yaml -------------------------------------------------------------------------------- /experiments/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/finetune.yaml -------------------------------------------------------------------------------- /experiments/hydra/launcher/slurm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/hydra/launcher/slurm.yaml -------------------------------------------------------------------------------- /experiments/task/franka.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/task/franka.yaml -------------------------------------------------------------------------------- /experiments/trainer/bc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/experiments/trainer/bc.yaml -------------------------------------------------------------------------------- /finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/finetune.py -------------------------------------------------------------------------------- /pretrained_networks_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/pretrained_networks_example.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SudeepDasari/data4robotics/HEAD/setup.py --------------------------------------------------------------------------------