Skip to content

airbert-vln/model-zoos

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

9 Commits
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿ˜๏ธ Airbert: In-domain Pretraining for Vision-and-Language Navigation ๐Ÿ˜๏ธ

MIT arXiv R2R 1st ICCV 2021 website

This repository stores some models trained in our experiments.

โŒจ๏ธ Downloading from the command line

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 pretrained on the BNB dataset

Model Description
airbert Airbert model pretrained on the BNB dataset

๐Ÿ‘ฝ External models used in our scripts

Model Description
vilbert ViLBERT model pretrained on Conceptual Captions
vlnbert VLN-BERT: vilbert fine-tuned on R2R

๐Ÿค– Finetuned models in discriminative setting

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

๐Ÿค– Finetuned models in generative setting

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

About

Model zoos for Airbert

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •