- Our model was trained and evaluated using the following package dependencies:
- Pytorch 1.9.1
- Python 3.6.12
-
Install Matterport3D simulators: follow instructions here.
-
Download object features here.
-
Download datasets of R2R and R4R here. It contains a datasets folder.
-
(Optional). Download the trained model here.
cd pretrain_src
CUDA_VISIBLE_DEVICES=0,1,2,3 bash run_r4r.sh 8001 # R4R
CUDA_VISIBLE_DEVICES=0,1,2,3 bash run_r2r.sh 8001 # R2R
CUDA_VISIBLE_DEVICES=0,1,2,3 bash scripts/r4r_b16.sh 8001 # R4R
CUDA_VISIBLE_DEVICES=0,1,2,3 bash scripts/r2r_b16.sh 8001 # R2R
Please see APAF_RxR/README.md
.