debug

mlcommons · Dec 7, 2023 · 0098e46 · 0098e46
1 parent 510e3d9
commit 0098e46
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/modeldiffs/fastmri_layernorm/compare.py b/tests/modeldiffs/fastmri_layernorm/compare.py
@@ -29,6 +29,7 @@ def sort_key(k):
   keys = sorted(sd.keys(), key=sort_key)
   c = 0
   for idx, k in enumerate(keys):
+    print(k)
     new_key = []
     layernorm = False
     for idx2, i in enumerate(k):