diff --git a/reading_machine/src/Classifier.cpp b/reading_machine/src/Classifier.cpp
index 2ea76708c9209ed217b127e40b4bef4e0677b00a..e91351e738de6e7fcfdf73abc486b85dddff9798 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