apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

only serialize weights

Browse Source
This commit is contained in:
stereomatchingkiss
2016-02-27 15:53:56 +08:00
parent f64a84b072
commit 9b672da91c
3 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/sparse_bias_layer.hpp
+1 -2
View File
@@ -136,8 +136,7 @@ class SparseBiasLayer
*/
template<typename Archive>
void Serialize(Archive& ar, const unsigned int /* version */)
{
ar & data::CreateNVP(lambda, "lambda");
{
ar & data::CreateNVP(weights, "weights");
}
src/mlpack/methods/ann/layer/sparse_input_layer.hpp
+1 -2
View File
@@ -138,8 +138,7 @@ class SparseInputLayer
*/
template<typename Archive>
void Serialize(Archive& ar, const unsigned int /* version */)
{
ar & data::CreateNVP(lambda, "lambda");
{
ar & data::CreateNVP(weights, "weights");
}
src/mlpack/methods/ann/layer/sparse_output_layer.hpp
+10 -13
View File
@@ -141,19 +141,7 @@ class SparseOutputLayer
double Rho() const
{
return rho;
}
/**
* Serialize the layer
*/
template<typename Archive>
void Serialize(Archive& ar, const unsigned int /* version */)
{
ar & data::CreateNVP(rhoCap, "rhoCap");
ar & data::CreateNVP(lambda, "lambda");
ar & data::CreateNVP(beta, "beta");
ar & data::CreateNVP(weights, "weights");
}
}
//! Get the weights.
OutputDataType const& Weights() const { return weights; }
@@ -184,6 +172,15 @@ class SparseOutputLayer
OutputDataType const& Gradient() const { return gradient; }
//! Modify the gradient.
OutputDataType& Gradient() { return gradient; }
/**
* Serialize the layer
*/
template<typename Archive>
void Serialize(Archive& ar, const unsigned int /* version */)
{
ar & data::CreateNVP(weights, "weights");
}
private:
//! Locally-stored number of input units.