Update README.md

README.md

@@ -20,13 +20,13 @@ class ProjectionModel(nn.Module):
         self.c_code_encoder = AutoModel.from_pretrained("microsoft/codebert-base")
         self.pseudocode_encoder = AutoModel.from_pretrained("microsoft/codebert-base")
 
-        # Projection network with 2 hidden layers
+        #Projection network with 2 hidden layers
         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),  #Second hidden layer
             nn.ReLU(),  # Second activation function
-            nn.Linear(config.hidden_dim, config.embedding_dim)  # Output layer projecting back to the original embedding space
+            nn.Linear(config.hidden_dim, config.embedding_dim)  #Output layer projecting back to the original embedding space
         )
 
     def forward(self, c_code_inputs, pseudocode_inputs):