def _build_projection_head(self):
    head = heads.DINOProjectionHead(512, 2048, 256, 2048, batch_norm=True)
    head.layers = heads.ProjectionHead([(512, 2048, nn.BatchNorm1d(2048), nn.GELU()), (2048, 256, None, None)]).layers
    return head
def _build_projection_head(self):
    head = heads.DINOProjectionHead(512, 2048, 256, 2048, batch_norm=True)
    head.layers = heads.ProjectionHead([(512, 2048, nn.BatchNorm1d(2048), nn.GELU()), (2048, 256, None, None)]).layers
    return head