Add idx 0

pomonam · Mar 19, 2024 · 2e84271 · 2e84271
1 parent 5aafa30
commit 2e84271
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/gpu_tests/ddp_test.py b/tests/gpu_tests/ddp_test.py
@@ -213,6 +213,10 @@ def test_lr_pairwise_scores(self) -> None:
         new_pairwise_scores = self.analyzer.load_pairwise_scores(scores_name="ddp_qb")
 
         if LOCAL_RANK == 0:
+            print(f"Previous score: {pairwise_scores[ALL_MODULE_NAME][0]}")
+            print(f"Previous shape: {pairwise_scores[ALL_MODULE_NAME].shape}")
+            print(f"New score: {new_pairwise_scores[ALL_MODULE_NAME][0]}")
+            print(f"New shape: {new_pairwise_scores[ALL_MODULE_NAME].shape}")
             print(f"Previous score: {pairwise_scores[ALL_MODULE_NAME][50]}")
             print(f"Previous shape: {pairwise_scores[ALL_MODULE_NAME].shape}")
             print(f"New score: {new_pairwise_scores[ALL_MODULE_NAME][50]}")