diff --git a/source/Layer.cpp b/source/Layer.cpp index 6e87c2e..c721962 100644 --- a/source/Layer.cpp +++ b/source/Layer.cpp @@ -52,7 +52,7 @@ bool Layer::loadWeights(const string &prjname) string filename = m_weightsFile; if (prjname.size() > 0) { - filename = prjname + string(".") + m_weightsFile; + filename = prjname + "." + m_weightsFile; } FILE *pFile = fopen(filename.c_str(),"r"); @@ -62,6 +62,7 @@ bool Layer::loadWeights(const string &prjname) return false; } + std::cout << "Importing weights for " << m_name << "." << to_string((int)m_id) << std::endl; int result = fscanf(pFile, "%d %d %d\n", &numVisibleX, &numVisibleY, &numHidden); if (result < 0) { @@ -107,10 +108,12 @@ bool Layer::loadWeights(const string &prjname) bool Layer::saveWeights(const string &prjname) { + int numHidden = m_bh.n_elem; + int numVisible = m_bv.n_elem; string filename = m_weightsFile; if (prjname.size() > 0) { - filename = prjname + string(".") + m_weightsFile; + filename = prjname + "." + m_weightsFile; } FILE *pFile = fopen(filename.c_str(),"w"); @@ -120,10 +123,9 @@ bool Layer::saveWeights(const string &prjname) return false; } - int numHidden = m_bh.n_elem; - int numVisible = m_bv.n_elem; - fprintf(pFile, "%d %d %d\n", (int)m_numVisibleX, (int)m_numVisibleY, numHidden); + std::cout << "Exporting weights for " << m_name << "." << to_string((int)m_id) << std::endl; + fprintf(pFile, "%d %d %d\n", (int)m_numVisibleX, (int)m_numVisibleY, numHidden); int i, j; for (i=0; i < numVisible; i++) diff --git a/source/Layer.hpp b/source/Layer.hpp index e5cbf23..722d369 100644 --- a/source/Layer.hpp +++ b/source/Layer.hpp @@ -37,6 +37,11 @@ public: arma::mat up_pass(const arma::mat& hidden); arma::mat down_pass(const arma::mat& visible); + std::string& name() + { + return m_name; + } + size_t id() { return m_id;