def METHOD_NAME(self, num_classes, global_pool='avg'):
    self.global_pool = SelectAdaptivePool2d(pool_type=global_pool)
    self.num_classes = num_classes
    self.classifier = nn.Linear(
        self.num_features * self.global_pool.feat_mult(),
        num_classes) if self.num_classes else None