From ae7ac3688627850db0756fc390cdef62b159cef7 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 9 Jul 2020 19:41:46 +0200
Subject: [PATCH] Moving nn to correct device
---
reading_machine/src/Classifier.cpp | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/reading_machine/src/Classifier.cpp b/reading_machine/src/Classifier.cpp
index 2ea7670..e91351e 100644
--- a/reading_machine/src/Classifier.cpp
+++ b/reading_machine/src/Classifier.cpp
@@ -80,6 +80,10 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std
resetOptimizer();
loadOptimizer();
}
+ else
+ {
+ getNN()->to(NeuralNetworkImpl::device);
+ }
}
int Classifier::getNbParameters() const
--
GitLab