Vinicius Avena

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...