~

src/two_tower_base_retrieval.py

@@ -251,7 +251,7 @@ def compute_training_loss(
         position: torch.Tensor,  # [B]
         labels: torch.Tensor  # [B, T]
     ) -> torch.Tensor:
-                # Compute the scores for every pair of user and item
+        # Compute the scores for every pair of user and item
         scores = torch.matmul(user_embedding, item_embeddings.t())  # [B, B]
 
         # You should either try to handle the popularity bias