diff --git a/reading_machine/src/Classifier.cpp b/reading_machine/src/Classifier.cpp index a9de8bbaabc1c3f20a2987e65cfb8094aaf470fc..ccc385288547de519d0170df54d8ec646824f260 100644 --- a/reading_machine/src/Classifier.cpp +++ b/reading_machine/src/Classifier.cpp @@ -83,9 +83,7 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std if (!train) { - fmt::print(stderr, "Before load on {}\n", NeuralNetworkImpl::getDevice() == torch::kCPU ? "cpu" : "gpu"); torch::load(getNN(), getBestFilename(), NeuralNetworkImpl::getDevice()); - fmt::print(stderr, "After load\n"); getNN()->registerEmbeddings(); getNN()->to(NeuralNetworkImpl::getDevice()); }