diff --git a/builtin_op_importers.cpp b/builtin_op_importers.cpp index 011d42a4..401db0a1 100644 --- a/builtin_op_importers.cpp +++ b/builtin_op_importers.cpp @@ -1703,6 +1703,7 @@ DEFINE_BUILTIN_OP_IMPORTER(Loop) const ::ONNX_NAMESPACE::GraphProto& body = attrs.get("body"); auto loop = ctx->network()->addLoop(); + loop->setName(getNodeName(node).c_str()); // Trip count and condition are optional inputs. nvinfer1::ITensor* tripLimit{nullptr}; if (inputs[0]) @@ -1732,7 +1733,6 @@ DEFINE_BUILTIN_OP_IMPORTER(Loop) ctx->loopTensors()[body.input(i).name()] = node.input(i); ctx->registerTensor(TensorOrWeights{stateVars.back()->getOutput(0)}, body.input(i).name()); } - ctx->registerLayer(stateVars.at(0), node.name()); // Loop body TRT_CHECK(onnx2trt::parseGraph(ctx, body)); diff --git a/onnx2trt_utils.cpp b/onnx2trt_utils.cpp index a7284706..69cd90d8 100644 --- a/onnx2trt_utils.cpp +++ b/onnx2trt_utils.cpp @@ -1325,6 +1325,10 @@ bool parseExternalWeights(IImporterContext* ctx, std::string file, std::string p { path.replace(slash + 1, path.size() - (slash + 1), file); } + else + { + path = file; + } std::ifstream relPathFile(path, std::ios::binary | std::ios::ate); if (!relPathFile) {