debugging

mlcommons · Dec 7, 2023 · 1fb0f66 · 1fb0f66
1 parent a03ebbc
commit 1fb0f66
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
@@ -53,6 +53,7 @@ def sort_key(k):
           i = i.replace('weight', 'kernel')
       new_key.append(i)
     new_key = tuple(new_key)
+    print(new_key)
     sd[new_key] = sd[k]
     del sd[k]
   return sd