Commit 80730f88 authored by Franck Dary's avatar Franck Dary
Browse files

Senting neural network to correct device

parent 70a1d8bc
......@@ -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()))
......
Supports Markdown
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