in vision/models.py [0:0]
def __init__(self, head): super(HeadAndEmbedding, self).__init__() self.head = head