debug

mlcommons · Dec 7, 2023 · 510e3d9 · 510e3d9
1 parent 99da8ba
commit 510e3d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modeldiffs/fastmri_layernorm/compare.py b/tests/modeldiffs/fastmri_layernorm/compare.py
@@ -49,7 +49,7 @@ def sort_key(k):
         if layernorm:
           i = i.replace('weight', 'scale')
         else:
-          i = i.replace('weight', 'dense')
+          i = i.replace('weight', 'kernel')
       new_key.append(i)
     new_key = tuple(new_key)
     sd[new_key] = sd[k]