From 1fb0f66f5750f46eaad550034d69e28c756ec9da Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Thu, 7 Dec 2023 05:47:06 +0000 Subject: [PATCH] debugging --- tests/modeldiffs/fastmri_layernorm/compare.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/modeldiffs/fastmri_layernorm/compare.py b/tests/modeldiffs/fastmri_layernorm/compare.py index a65c91b4b..2fc93e311 100644 --- a/tests/modeldiffs/fastmri_layernorm/compare.py +++ b/tests/modeldiffs/fastmri_layernorm/compare.py @@ -53,6 +53,7 @@ def sort_key(k): i = i.replace('weight', 'kernel') new_key.append(i) new_key = tuple(new_key) + print(new_key) sd[new_key] = sd[k] del sd[k] return sd