├── README.md ├── agents ├── __init__.py ├── agent.py ├── navigation_agent.py └── random_agent.py ├── datasets ├── __init__.py ├── base_controller.py ├── constants.py ├── data.py ├── environment.py ├── offline_controller_with_small_rotation.py └── target_util.py ├── episodes ├── __init__.py ├── basic_episode.py ├── episode.py └── test_val_episode.py ├── full_eval.py ├── main.py ├── main_eval.py ├── main_pretraining.py ├── models ├── __init__.py ├── basemodel.py ├── model_io.py ├── pretrainedvisualtransformer.py ├── transformer.py └── vtnetmodel.py ├── optimizers ├── __init__.py ├── shared_RMS_prop.py └── shared_adam.py ├── requirements.txt ├── runners ├── __init__.py ├── a3c_train.py ├── a3c_val.py ├── pretrain_engine.py └── train_util.py ├── single_eval.py ├── test_val_split ├── bathroom_test_22.pkl ├── bathroom_test_val_22.pkl ├── bathroom_val_22.pkl ├── bedroom_test_22.pkl ├── bedroom_test_val_22.pkl ├── bedroom_val_22.pkl ├── kitchen_test_22.pkl ├── kitchen_test_val_22.pkl ├── kitchen_val_22.pkl ├── living_room_test_22.pkl ├── living_room_test_val_22.pkl └── living_room_val_22.pkl └── utils ├── __init__.py ├── action_util.py ├── class_finder.py ├── command_parser.py ├── data_utils.py ├── misc.py ├── misc_util.py ├── model_util.py └── pretrain_util.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/README.md -------------------------------------------------------------------------------- /agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/agents/__init__.py -------------------------------------------------------------------------------- /agents/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/agents/agent.py -------------------------------------------------------------------------------- /agents/navigation_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/agents/navigation_agent.py -------------------------------------------------------------------------------- /agents/random_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/agents/random_agent.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/base_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/base_controller.py -------------------------------------------------------------------------------- /datasets/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/constants.py -------------------------------------------------------------------------------- /datasets/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/data.py -------------------------------------------------------------------------------- /datasets/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/environment.py -------------------------------------------------------------------------------- /datasets/offline_controller_with_small_rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/offline_controller_with_small_rotation.py -------------------------------------------------------------------------------- /datasets/target_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/datasets/target_util.py -------------------------------------------------------------------------------- /episodes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/episodes/__init__.py -------------------------------------------------------------------------------- /episodes/basic_episode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/episodes/basic_episode.py -------------------------------------------------------------------------------- /episodes/episode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/episodes/episode.py -------------------------------------------------------------------------------- /episodes/test_val_episode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/episodes/test_val_episode.py -------------------------------------------------------------------------------- /full_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/full_eval.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/main.py -------------------------------------------------------------------------------- /main_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/main_eval.py -------------------------------------------------------------------------------- /main_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/main_pretraining.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/basemodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/basemodel.py -------------------------------------------------------------------------------- /models/model_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/model_io.py -------------------------------------------------------------------------------- /models/pretrainedvisualtransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/pretrainedvisualtransformer.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/vtnetmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/models/vtnetmodel.py -------------------------------------------------------------------------------- /optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/optimizers/__init__.py -------------------------------------------------------------------------------- /optimizers/shared_RMS_prop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/optimizers/shared_RMS_prop.py -------------------------------------------------------------------------------- /optimizers/shared_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/optimizers/shared_adam.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/runners/__init__.py -------------------------------------------------------------------------------- /runners/a3c_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/runners/a3c_train.py -------------------------------------------------------------------------------- /runners/a3c_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/runners/a3c_val.py -------------------------------------------------------------------------------- /runners/pretrain_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/runners/pretrain_engine.py -------------------------------------------------------------------------------- /runners/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/runners/train_util.py -------------------------------------------------------------------------------- /single_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/single_eval.py -------------------------------------------------------------------------------- /test_val_split/bathroom_test_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bathroom_test_22.pkl -------------------------------------------------------------------------------- /test_val_split/bathroom_test_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bathroom_test_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/bathroom_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bathroom_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/bedroom_test_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bedroom_test_22.pkl -------------------------------------------------------------------------------- /test_val_split/bedroom_test_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bedroom_test_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/bedroom_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/bedroom_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/kitchen_test_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/kitchen_test_22.pkl -------------------------------------------------------------------------------- /test_val_split/kitchen_test_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/kitchen_test_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/kitchen_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/kitchen_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/living_room_test_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/living_room_test_22.pkl -------------------------------------------------------------------------------- /test_val_split/living_room_test_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/living_room_test_val_22.pkl -------------------------------------------------------------------------------- /test_val_split/living_room_val_22.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/test_val_split/living_room_val_22.pkl -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/action_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/action_util.py -------------------------------------------------------------------------------- /utils/class_finder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/class_finder.py -------------------------------------------------------------------------------- /utils/command_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/command_parser.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/misc_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/misc_util.py -------------------------------------------------------------------------------- /utils/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/model_util.py -------------------------------------------------------------------------------- /utils/pretrain_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xiaobaishu0097/ICLR_VTNet/HEAD/utils/pretrain_util.py --------------------------------------------------------------------------------