diff --git a/src/qonnx/core/modelwrapper.py b/src/qonnx/core/modelwrapper.py index c7b2c676..fffb7424 100644 --- a/src/qonnx/core/modelwrapper.py +++ b/src/qonnx/core/modelwrapper.py @@ -44,6 +44,7 @@ from qonnx.transformation.general import ( RemoveStaticGraphInputs, RemoveUnusedTensors, + GiveUniqueParameterTensors, SortCommutativeInputsInitializerLast, SortGraph, ) @@ -209,6 +210,7 @@ def cleanup(self): RemoveStaticGraphInputs(), SortGraph(), SortCommutativeInputsInitializerLast(), + GiveUniqueParameterTensors(), ] for trn in cleanup_transforms: transformed_model = transformed_model.transform(trn, cleanup=False, make_deepcopy=False)