Commit ae7ac368 authored by Franck Dary's avatar Franck Dary
Browse files

Moving nn to correct device

parent 761ea87c
......@@ -80,6 +80,10 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std
resetOptimizer();
loadOptimizer();
}
else
{
getNN()->to(NeuralNetworkImpl::device);
}
}
int Classifier::getNbParameters() const
......
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