From 8b49fce5013bc5d34efd9c91279454153be43cfb Mon Sep 17 00:00:00 2001 From: Moritz Reiber Date: Thu, 18 Apr 2024 11:25:21 +0000 Subject: [PATCH] Update dense weight id --- hannah/nas/functional_operators/operators.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hannah/nas/functional_operators/operators.py b/hannah/nas/functional_operators/operators.py index 5dcc06b0..d9fe539b 100644 --- a/hannah/nas/functional_operators/operators.py +++ b/hannah/nas/functional_operators/operators.py @@ -187,6 +187,7 @@ def __call__(self, *operands) -> Any: new_linear = super().__call__(*operands) new_linear.in_features = operands[1].shape()[0] new_linear.out_features = operands[1].shape()[1] + operands[1].id = f"{new_linear.id}.{operands[1].id}" return new_linear def shape_fun(self):