From f70dd49f9c6e60038be4c0bb07d962903c8bbc31 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Sat, 18 Nov 2023 03:52:01 +0000 Subject: [PATCH] debug --- tests/modeldiffs/criteo1tb_resnet/compare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/modeldiffs/criteo1tb_resnet/compare.py b/tests/modeldiffs/criteo1tb_resnet/compare.py index 6eb7583c4..b5b0cfb62 100644 --- a/tests/modeldiffs/criteo1tb_resnet/compare.py +++ b/tests/modeldiffs/criteo1tb_resnet/compare.py @@ -29,13 +29,13 @@ def key_transform(k): if 'Embedding' in i: return ('embedding_table',) if 'ResNetBlock' in i: - i = i.replace('ResNetBlock', 'Dense') name, count = i.split('_') resnet_count = resnet_count + 1 continue if 'Linear' in i: i = i.replace('Linear', 'Dense') name, count = i.split('_') + print(resnet_count) i = name + '_' + str(s_count * 3 + int(resnet_count)) elif 'weight' in i: i = i.replace('weight', 'kernel')