From 991e94f3d7463aee466cbad0c2c36b9a93aaf229 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Wed, 8 Apr 2020 23:25:30 +0200
Subject: [PATCH] setting neural network to correct device
---
reading_machine/src/Classifier.cpp | 2 ++
1 file changed, 2 insertions(+)
diff --git a/reading_machine/src/Classifier.cpp b/reading_machine/src/Classifier.cpp
index af823b7..a51ad08 100644
--- a/reading_machine/src/Classifier.cpp
+++ b/reading_machine/src/Classifier.cpp
@@ -53,6 +53,8 @@ void Classifier::initNeuralNetwork(const std::vector<std::string> & definition)
initLSTM(definition, curIndex);
else
util::myThrow(fmt::format("Unknown network type '{}', available types are 'Random, LSTM'", networkType));
+
+ this->nn->to(NeuralNetworkImpl::device);
}
void Classifier::initLSTM(const std::vector<std::string> & definition, std::size_t & curIndex)
--
GitLab