├── .gitignore ├── README.md ├── configs └── bert_config.json ├── datasets ├── data_module.py └── pretrain_dataset.py ├── main.py ├── models ├── bert.py ├── encoder.py ├── irl_model.py ├── resnet.py └── vit.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/README.md -------------------------------------------------------------------------------- /configs/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/configs/bert_config.json -------------------------------------------------------------------------------- /datasets/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/datasets/data_module.py -------------------------------------------------------------------------------- /datasets/pretrain_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/datasets/pretrain_dataset.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/main.py -------------------------------------------------------------------------------- /models/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/models/bert.py -------------------------------------------------------------------------------- /models/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/models/encoder.py -------------------------------------------------------------------------------- /models/irl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/models/irl_model.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/models/vit.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASGMVLP/ASGMVLP_CODE/HEAD/requirements.txt --------------------------------------------------------------------------------