MLP Dimension on ResNet Module

[viniavena]

Hi there, in the models/resnet_simclr.py we have the following:

dim_mlp = self.backbone.fc.in_features self.backbone.fc = nn.Sequential(nn.Linear(dim_mlp, dim_mlp), nn.ReLU(), self.backbone.fc)

Shouldn't the Linear be: dim_mlp, 128

Threfore we get the 128 dimension hidden vector used on the paper.