Update README.md

README.md

@@ -24,7 +24,7 @@ class ProjectionModel(nn.Module):
         self.projection = nn.Sequential(
             nn.Linear(config.embedding_dim, config.hidden_dim),
             nn.ReLU(),  # First activation function
-            nn.Linear(config.hidden_dim, config.hidden_dim),  #Second hidden layer
+            nn.Linear(config.hidden_dim, config.hidden_dim),  #Hidden layer
             nn.ReLU(),  # Second activation function
             nn.Linear(config.hidden_dim, config.embedding_dim)  #Output layer projecting back to the original embedding space
         )