Skip to content
Snippets Groups Projects
Commit 5acd244b authored by Franck Dary's avatar Franck Dary
Browse files

Removed debug print

parent 4584da0f
Branches
No related tags found
No related merge requests found
...@@ -83,9 +83,7 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std ...@@ -83,9 +83,7 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std
if (!train) if (!train)
{ {
fmt::print(stderr, "Before load on {}\n", NeuralNetworkImpl::getDevice() == torch::kCPU ? "cpu" : "gpu");
torch::load(getNN(), getBestFilename(), NeuralNetworkImpl::getDevice()); torch::load(getNN(), getBestFilename(), NeuralNetworkImpl::getDevice());
fmt::print(stderr, "After load\n");
getNN()->registerEmbeddings(); getNN()->registerEmbeddings();
getNN()->to(NeuralNetworkImpl::getDevice()); getNN()->to(NeuralNetworkImpl::getDevice());
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment