This repository stores some models trained in our experiments.
We stored our models on Google Drive, as the service provides a storage of 15 GB for free.
You can use the great gdown script for downloading the models:
pip install gdown
gdown [link to Google Drive]We also provide a Makefile to help you:
# Download everything
make all
# Download a specific model
make airbert-r2rRSA
# Get all commands
make help| Model | Description |
|---|---|
| airbert | Airbert model pretrained on the BNB dataset |
| Model | Description |
|---|---|
| vilbert | ViLBERT model pretrained on Conceptual Captions |
| vlnbert | VLN-BERT: vilbert fine-tuned on R2R |
| Model | Description |
|---|---|
| airbert-r2rRS | airbert fine-tuned on R2R with the shuffling loss |
| airbert-r2rRSA | airbert fine-tuned on R2R with the shuffling loss + speaker data |
| Model | Description |
|---|---|
| REVERIE | Recurrent VLN-BERT for remote referring expression with pretrained Airbert as backbone |
| R2R | Recurrent VLN-BERT for vision-and-language navigation with pretrained Airbert as backbone |