[mlpack-git] [mlpack] Serialization of ann (#513)
Joseph Mariadassou
notifications at github.com
Fri Mar 4 15:33:49 EST 2016
Serialization is not complete AFAIK: The following test fails (100% error):
```
auto net = CreateVanillaNetwork();
RMSprop<decltype(net)> opt(net, 0.01, 0.88, 1e-8, 10 * input.n_slices, 0);
net.Train(input, Y, opt);
data::Save("LearntMachine.xml", "ConvNetwork", net, mlpack::data::xml );
size_t error = 0;
{
arma::mat prediction;
net.Predict(input, prediction);
...
}
}
auto net2 = CopyVanillaNetwork();
data::Load("LearntMachine.xml", "ConvNetwork", net2, mlpack::data::xml);
{
arma::mat prediction;
net2.Predict(input, prediction);
...
}
```
Note that `net` computes the values as expected. But `net2` fails with 100% error. This is partly because the weights have not been serialized. But adding `network` and `outputLayer` to CNN::Serialize did not help.
---
Reply to this email directly or view it on GitHub:
https://github.com/mlpack/mlpack/issues/513#issuecomment-192457197
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160304/5c05a713/attachment.html>
More information about the mlpack-git
mailing list