Commit 5acd244b authored by Franck Dary's avatar Franck Dary
Browse files

Removed debug print

parent 4584da0f
......@@ -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());
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment