debug

mlcommons · Nov 22, 2023 · c44b5bd · c44b5bd
1 parent 8243820
commit c44b5bd
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
@@ -19,7 +19,7 @@ def key_transform(k):
   new_key = []
   s_count = None
   print('key')
-  print(key)
+  print(k)
   for i in k:
     if 'Sequential' in i:
       s_count = int(i.split('_')[1])