From c5efaf02b7f62df13f22da7c07757aa90762b53b Mon Sep 17 00:00:00 2001 From: nihuini Date: Thu, 23 Dec 2021 14:28:36 +0800 Subject: [PATCH] unique attribute identifier --- tools/pnnx/src/ir.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/pnnx/src/ir.cpp b/tools/pnnx/src/ir.cpp index 58a007fec..53e997909 100644 --- a/tools/pnnx/src/ir.cpp +++ b/tools/pnnx/src/ir.cpp @@ -1341,11 +1341,11 @@ int Graph::python(const std::string& pypath, const std::string& pnnxbinpath) if (is_running_mean_var) { - fprintf(pyfp, " self.%s = self.load_pnnx_bin_as_tensor(archive, '%s.%s', (", key.c_str(), sanitize_identifier(op->name).c_str(), key.c_str()); + fprintf(pyfp, " self.%s_%s = self.load_pnnx_bin_as_tensor(archive, '%s.%s', (", sanitize_identifier(op->name).c_str(), key.c_str(), sanitize_identifier(op->name).c_str(), key.c_str()); } else { - fprintf(pyfp, " self.%s = self.load_pnnx_bin_as_parameter(archive, '%s.%s', (", key.c_str(), sanitize_identifier(op->name).c_str(), key.c_str()); + fprintf(pyfp, " self.%s_%s = self.load_pnnx_bin_as_parameter(archive, '%s.%s', (", sanitize_identifier(op->name).c_str(), key.c_str(), sanitize_identifier(op->name).c_str(), key.c_str()); } for (size_t i = 0; i < attr.shape.size(); i++) @@ -1420,7 +1420,7 @@ int Graph::python(const std::string& pypath, const std::string& pnnxbinpath) else if (op->type == "pnnx.Attribute") { const std::string& key = op->attrs.begin()->first; - fprintf(pyfp, "v_%s = self.%s\n", sanitize_identifier(op->outputs[0]->name).c_str(), key.c_str()); + fprintf(pyfp, "v_%s = self.%s_%s\n", sanitize_identifier(op->outputs[0]->name).c_str(), sanitize_identifier(op->name).c_str(), key.c_str()); } else if (op->type == "Tensor.slice") {