From ff3894a31c6ab51e00a1040fd7d47e9e2304017d Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 9 Apr 2020 15:26:09 +0200
Subject: [PATCH] Fixed bug in MLP init
---
reading_machine/src/Classifier.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/reading_machine/src/Classifier.cpp b/reading_machine/src/Classifier.cpp
index 089ee2b..27a4ef1 100644
--- a/reading_machine/src/Classifier.cpp
+++ b/reading_machine/src/Classifier.cpp
@@ -169,7 +169,7 @@ void Classifier::initLSTM(const std::vector<std::string> & definition, std::size
if (params.size() % 2)
util::myThrow("MLP must have even number of parameters");
for (unsigned int i = 0; i < params.size()/2; i++)
- mlp.emplace_back(std::make_pair(std::stoi(params[i]), std::stof(params[i+1])));
+ mlp.emplace_back(std::make_pair(std::stoi(params[2*i]), std::stof(params[2*i+1])));
curIndex++;
}))
util::myThrow(fmt::format("Invalid line '{}', expected '{}'\n", curIndex < definition.size() ? definition[curIndex] : "", "(MLP :) {hidden1 dropout1 hidden2 dropout2...}"));
--
GitLab