fix bug for dimensions

moco/builder.py

@@ -17,7 +17,6 @@ class MoCo(nn.Module):
         """
         dim: feature dimension (default: 256)
         mlp_dim: hidden dimension in MLPs (default: 4096)
-        m: moco momentum of updating momentum encoder (default: 0.99)
         T: softmax temperature (default: 1.0)
         """
         super(MoCo, self).__init__()
@@ -111,7 +110,7 @@ class MoCo_ResNet(MoCo):
 
 
 class MoCo_ViT(MoCo):
-    def _build_projector_and_predictor_mlps(self, base_encoder, dim=256, mlp_dim=4096):
+    def _build_projector_and_predictor_mlps(self, dim, mlp_dim):
         hidden_dim = self.base_encoder.head.weight.shape[1]
         del self.base_encoder.head, self.momentum_encoder.head # remove original fc layer