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