Senting neural network to correct device

reading_machine/src/Classifier.cpp

@@ -66,6 +66,8 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std
   getNN()->loadDicts(path);
   getNN()->registerEmbeddings();
 
+  getNN()->to(NeuralNetworkImpl::device);
+
   if (!train)
     torch::load(getNN(), getBestFilename());
   else if (std::filesystem::exists(getLastFilename()))