setting neural network to correct device

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)