Method __init__

SwissArmyTransformer/sat/model/official/vit_model.py:96–98

(self, hidden_size, num_classes)

class ClsMixin(BaseMixin):
    def __init__(self, hidden_size, num_classes):
        super().__init__()
        self.classifier = nn.Linear(hidden_size, num_classes)

    def final_forward(self, logits, **kw_args):
        logits = self.classifier(logits[:, 0])