Skip to content
Snippets Groups Projects
Commit 222ee14f authored by Jerome Hue's avatar Jerome Hue Committed by Olivier BICHLER
Browse files

chore: reorder parameters of Leaky constructor

parent be665712
No related branches found
No related tags found
2 merge requests!318[Upd] release verision 0.5.0,!283[Add] Operator Leaky_MetaOperator
......@@ -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
......
......@@ -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") = "");
}
......
......@@ -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");
......
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