From 9550c15f8bb67e4cdc82ed41d069087167cc13e4 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Wed, 22 Nov 2023 04:06:27 +0000 Subject: [PATCH] debug --- tests/modeldiffs/criteo1tb_layernorm/compare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/modeldiffs/criteo1tb_layernorm/compare.py b/tests/modeldiffs/criteo1tb_layernorm/compare.py index e56d65126..a8ee234aa 100644 --- 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)