debug

mlcommons · Nov 18, 2023 · 271412d · 271412d
1 parent 0c7a864
commit 271412d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/modeldiffs/criteo1tb/compare.py b/tests/modeldiffs/criteo1tb/compare.py
@@ -20,7 +20,7 @@ def key_transform(k):
   new_key = []
   s_count = None
   for i in k:
-    print(i)
+    print(f'in transform: {i}')
     if 'Sequential' in i:
       s_count = int(i.split('_')[1])
       continue
@@ -34,7 +34,7 @@ def key_transform(k):
       i = i.replace('weight', 'kernel')
 
     new_key.append(i)
-    print(i)
+    print(f'out transform: {i}')
   return tuple(new_key)
 
 

diff --git a/tests/modeldiffs/criteo1tb_resnet/compare.py b/tests/modeldiffs/criteo1tb_resnet/compare.py
@@ -21,7 +21,7 @@ def key_transform(k):
   s_count = None
   print(k)
   for i in k:
-    print(i)
+    print(f'in transform {i}')
     if 'Sequential' in i:
       s_count = int(i.split('_')[1])
       continue
@@ -37,7 +37,7 @@ def key_transform(k):
       i = name + '_' + str(s_count * 3 + int(count))
     elif 'weight' in i:
       i = i.replace('weight', 'kernel')
-    print(i)
+    print(f'out transform {i}')
     new_key.append(i)
   return tuple(new_key)