Skip to content
Snippets Groups Projects
Commit c3af0b69 authored by Lucas Lopez's avatar Lucas Lopez
Browse files

Apply 2 suggestion(s) to 2 file(s)


Co-authored-by: default avatarCyril Moineau <cyril.moineau@cea.fr>
parent 808cadc8
No related branches found
No related tags found
2 merge requests!47Draft: Duplication of !41,!41Draft: Quantized Export to ONNX: Recipe for Transforming an Aidge Model into an ONNX-Quantized Friendly Model
......@@ -34,5 +34,5 @@ def freeze_weights(graphview: aidge_core.GraphView, all_producers: bool = False)
while(parent_node.type() != "Producer"):
parent_node = parent_node.get_parent(0)
if parent_node is None:
raise ValueError ("error, producer not found")
raise RuntimeError(f"Could not find a parent producer for node {node.name()}")
parent_node.get_operator().attr.set_attr("constant",True)
......@@ -21,7 +21,7 @@ namespace Aidge {
void init_ONNXRecipes(py::module &m) {
m.def("quantize_matching_to_export", &quantizeMatchingtoExport, py::arg("graphView"), py::arg("qop")=true,py::arg("fold_weights")=true);
m.def("quantize_matching_to_export", &quantizeMatchingtoExport, py::arg("graph_view"), py::arg("qop")=true, py::arg("fold_weights")=true);
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment