Skip to content

Commit

Permalink
formatted code using make format
Browse files Browse the repository at this point in the history
  • Loading branch information
LukeSutor committed Feb 4, 2024
1 parent 3fdd610 commit 34aa0ec
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions lightly/models/modules/heads.py
Original file line number Diff line number Diff line change
Expand Up @@ -641,13 +641,13 @@ def forward(self, x: Tensor) -> Tensor:
x = nn.functional.normalize(x, dim=-1, p=2)
x = self.last_layer(x)
return x


class MMCRProjectionHead(ProjectionHead):
"""Projection head used for MMCR.
"Following Chen et al. (14), we append a small perceptron to the output
of the average pooling layer of the ResNet so that zi = g(h(xi)), where
"Following Chen et al. (14), we append a small perceptron to the output
of the average pooling layer of the ResNet so that zi = g(h(xi)), where
h is the ResNet and g is the MLP." [0]
- [0]: MMCR, 2023, https://arxiv.org/abs/2303.03307
Expand Down

0 comments on commit 34aa0ec

Please sign in to comment.