src/transformers/models/flava/modeling_flava.py/FlavaITMHead/__init__
class FlavaITMHead:    def __init__(self, config):
        super().__init__()
        self.config = config
        self.pooler = FlavaPooler(config)
        self.seq_relationship = nn.Linear(config.hidden_size, 2)