debug

mlcommons · Nov 22, 2023 · 9550c15 · 9550c15
1 parent a140fc0
commit 9550c15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modeldiffs/criteo1tb_layernorm/compare.py b/tests/modeldiffs/criteo1tb_layernorm/compare.py
@@ -39,8 +39,8 @@ def key_transform(k):
         i = name + '_' + str(3)
     elif 'weight' in i:
       i = i.replace('weight', 'kernel')
-
     new_key.append(i)
+  print(new_key)
   return tuple(new_key)