diff --git a/src/finn/core/modelwrapper.py b/src/finn/core/modelwrapper.py index 1f8cef7888cf2404f4dd9627b0b5cbf40189ac65..cb70545f31e8776af33ebb936c5ebc3c6f8cbf92 100644 --- a/src/finn/core/modelwrapper.py +++ b/src/finn/core/modelwrapper.py @@ -183,6 +183,9 @@ class ModelWrapper: # sweep over value_info if util.get_by_name(graph.value_info, old_name) is not None: util.get_by_name(graph.value_info, old_name).name = new_name + # sweep over initializers + if util.get_by_name(graph.initializer, old_name) is not None: + util.get_by_name(graph.initializer, old_name).name = new_name # sweep over quantization annotations if ( util.get_by_name(graph.quantization_annotation, old_name, "tensor_name")