src/transformers/models/bridgetower/modeling_bridgetower.py/BridgeTowerITMHead/__init__
class BridgeTowerITMHead:    def __init__(self, hidden_size):
        super().__init__()
        self.fc = nn.Linear(hidden_size, 2)