diff --git a/include/aidge/operator/MetaOperatorDefs.hpp b/include/aidge/operator/MetaOperatorDefs.hpp index f80dd38f124a1730519ce6c1e0a4cefcfa8a359a..c9de36b1f032f986f6bd4461198ecff77d64d1f7 100644 --- a/include/aidge/operator/MetaOperatorDefs.hpp +++ b/include/aidge/operator/MetaOperatorDefs.hpp @@ -213,8 +213,8 @@ std::shared_ptr<MetaOperator_Op> LSTM_Op(DimSize_t seq_length); std::shared_ptr<MetaOperator_Op> LeakyOp(); std::shared_ptr<Node> Leaky(const int nbTimeSteps, - const float threshold, const float beta, + const float threshold = 1.0, const std::string &name = ""); } // namespace Aidge diff --git a/python_binding/operator/pybind_MetaOperatorDefs.cpp b/python_binding/operator/pybind_MetaOperatorDefs.cpp index 90edfe791a5a1c87fc18b961eded32358e407c5f..b2811fbaab2b6cd33dc2b105f0044cd8a5edbbc7 100644 --- a/python_binding/operator/pybind_MetaOperatorDefs.cpp +++ b/python_binding/operator/pybind_MetaOperatorDefs.cpp @@ -182,8 +182,8 @@ void declare_LSTMOp(py::module &m) { void declare_LeakyOp(py::module &m) { m.def("Leaky", &Leaky, py::arg("nb_timesteps"), - py::arg("threshold"), py::arg("beta"), + py::arg("threshold") = 1.0, py::arg("name") = ""); } diff --git a/src/operator/MetaOperatorDefs/Leaky.cpp b/src/operator/MetaOperatorDefs/Leaky.cpp index 17296d1305db1a832e8e055916a95c7ea0a92f00..5513323fd7a003af029bf1bd454b9a1bf4185144 100644 --- a/src/operator/MetaOperatorDefs/Leaky.cpp +++ b/src/operator/MetaOperatorDefs/Leaky.cpp @@ -14,8 +14,8 @@ constexpr auto memorizeOpDataOutputRecIndex = 1; constexpr auto memorizeOpDataOutputIndex = 0; std::shared_ptr<Node> Leaky(const int nbTimeSteps, - const float threshold, const float beta, + const float threshold, const std::string &name) { Log::warn("! Lots of parameters are hardcoded");