├── .gitattributes ├── CODEOWNERS ├── CONTRIBUTING-ARCHIVED.md ├── Dockerfile ├── GPU4.yaml ├── LICENSE ├── README.md ├── model_caption.png ├── pred ├── test_full_ranks_model5.3.355_listnet_s42.json └── test_full_ranks_model_new_assemble_bigger_normalize.json ├── pytorch_pretrained_bert ├── __init__.py ├── __main__.py ├── file_utils.py ├── loss.py ├── modeling.py ├── optimization.py ├── optimization_fp16.py ├── rank_loss.py └── tokenization.py ├── sh ├── finetune_v1.0_disc_dense_g4.sh ├── finetune_v1.0_disc_g4.sh ├── finetune_v1.0_gen_g4.sh ├── pretrain_v1.0_mlm_nsp_g2.sh ├── test_v1.0_disc_test.sh ├── test_v1.0_disc_val.sh └── test_v1.0_gen_val.sh └── vdbert ├── __init__.py ├── data_parallel.py ├── ensemble_rank.py ├── extract_visdial1_feature.py ├── loader_utils.py ├── metrics.py ├── seq2seq_loader.py ├── test_visdial_disc_test.py ├── test_visdial_disc_val.py ├── test_visdial_gen_val.py ├── train_visdial.py └── train_visdial_disc_dense.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/.gitattributes -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CONTRIBUTING-ARCHIVED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/CONTRIBUTING-ARCHIVED.md -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/Dockerfile -------------------------------------------------------------------------------- /GPU4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/GPU4.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/README.md -------------------------------------------------------------------------------- /model_caption.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/model_caption.png -------------------------------------------------------------------------------- /pred/test_full_ranks_model5.3.355_listnet_s42.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pred/test_full_ranks_model5.3.355_listnet_s42.json -------------------------------------------------------------------------------- /pred/test_full_ranks_model_new_assemble_bigger_normalize.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pred/test_full_ranks_model_new_assemble_bigger_normalize.json -------------------------------------------------------------------------------- /pytorch_pretrained_bert/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/__init__.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/__main__.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/file_utils.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/loss.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/modeling.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/optimization.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/optimization_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/optimization_fp16.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/rank_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/rank_loss.py -------------------------------------------------------------------------------- /pytorch_pretrained_bert/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/pytorch_pretrained_bert/tokenization.py -------------------------------------------------------------------------------- /sh/finetune_v1.0_disc_dense_g4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/finetune_v1.0_disc_dense_g4.sh -------------------------------------------------------------------------------- /sh/finetune_v1.0_disc_g4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/finetune_v1.0_disc_g4.sh -------------------------------------------------------------------------------- /sh/finetune_v1.0_gen_g4.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/finetune_v1.0_gen_g4.sh -------------------------------------------------------------------------------- /sh/pretrain_v1.0_mlm_nsp_g2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/pretrain_v1.0_mlm_nsp_g2.sh -------------------------------------------------------------------------------- /sh/test_v1.0_disc_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/test_v1.0_disc_test.sh -------------------------------------------------------------------------------- /sh/test_v1.0_disc_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/test_v1.0_disc_val.sh -------------------------------------------------------------------------------- /sh/test_v1.0_gen_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/sh/test_v1.0_gen_val.sh -------------------------------------------------------------------------------- /vdbert/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vdbert/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/data_parallel.py -------------------------------------------------------------------------------- /vdbert/ensemble_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/ensemble_rank.py -------------------------------------------------------------------------------- /vdbert/extract_visdial1_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/extract_visdial1_feature.py -------------------------------------------------------------------------------- /vdbert/loader_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/loader_utils.py -------------------------------------------------------------------------------- /vdbert/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/metrics.py -------------------------------------------------------------------------------- /vdbert/seq2seq_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/seq2seq_loader.py -------------------------------------------------------------------------------- /vdbert/test_visdial_disc_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/test_visdial_disc_test.py -------------------------------------------------------------------------------- /vdbert/test_visdial_disc_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/test_visdial_disc_val.py -------------------------------------------------------------------------------- /vdbert/test_visdial_gen_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/test_visdial_gen_val.py -------------------------------------------------------------------------------- /vdbert/train_visdial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/train_visdial.py -------------------------------------------------------------------------------- /vdbert/train_visdial_disc_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/salesforce/VD-BERT/HEAD/vdbert/train_visdial_disc_dense.py --------------------------------------------------------------------------------