From cf0ef7856b1016e52f0967f2d70dee887d73b8d6 Mon Sep 17 00:00:00 2001
From: Carl Philipp Klemm <philipp@uvos.xyz>
Date: Mon, 6 May 2024 18:10:43 +0200
Subject: [PATCH] support getParameterNames
---
src/main.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/main.cpp b/src/main.cpp
index 4cb011e..29158de 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -57,6 +57,7 @@ PYBIND11_MODULE(_core, m)
.def("getTorchScript", &Model::getTorchScript)
.def("getCompiledFunctionName", &Model::getCompiledFunctionName)
.def("getFlatParameters", &Model::getFlatParameters)
+ .def("getParameterNames", &Model::getParameterNames)
.def("getParameterCount", &Model::getParameterCount)
.def("getRecommendedParamIndices", &Model::getRecommendedParamIndices)
.def("__repr__", &Model::getModelStr);
--
GitLab