def _alexnet_replace_fc()

in vision/models.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

def _alexnet_replace_fc(model):
    model.classifier = HeadAndEmbedding(model.classifier)
    return model