https://github.com/declare-lab/MSA-Robustness/blob/2466d33c767411f6490c8c022666e8d9167696b9/Multimodal-Infomax/src/model.py#L8