From 510e3d92486023247ecf8a1bfd9836525c98c0ea Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Thu, 7 Dec 2023 04:20:56 +0000 Subject: [PATCH] debug --- tests/modeldiffs/fastmri_layernorm/compare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/modeldiffs/fastmri_layernorm/compare.py b/tests/modeldiffs/fastmri_layernorm/compare.py index cf0194320..cb51f6bd9 100644 --- a/tests/modeldiffs/fastmri_layernorm/compare.py +++ b/tests/modeldiffs/fastmri_layernorm/compare.py @@ -49,7 +49,7 @@ def sort_key(k): if layernorm: i = i.replace('weight', 'scale') else: - i = i.replace('weight', 'dense') + i = i.replace('weight', 'kernel') new_key.append(i) new_key = tuple(new_key) sd[new_key] = sd[k]