jens/Rbm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

- Layer:don't store weight layout info. It's already stored in prj-file

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@772 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-10 11:37:00 +00:00
parent 648d8a002e
commit 9f73824bc4
1 changed files with 3 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Layer.cpp
+3 -12
View File
@@ -47,10 +47,6 @@ Layer::~Layer()
bool Layer::loadWeights(const string &prjname) bool Layer::loadWeights(const string &prjname)
{ {
int numVisibleX;
int numVisibleY;
int numHidden;
string filename = m_weightsFile; string filename = m_weightsFile;
if (prjname.size() > 0) if (prjname.size() > 0)
{ {
@@ -65,16 +61,12 @@ bool Layer::loadWeights(const string &prjname)
} }
std::cout << "Importing weights for " << m_name << "." << to_string((int)m_id) << std::endl; 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)
{
return false;
}
size_t numVisible = numVisibleX*numVisibleY;
size_t numVisible = Rbm::numVisible();
size_t numHidden = Rbm::numHidden();
int i, j; int i, j;
float v; float v;
int result;
arma::mat _bv(1, numVisible); arma::mat _bv(1, numVisible);
for (i=0; i < numVisible; i++) for (i=0; i < numVisible; i++)
{ {
@@ -131,7 +123,6 @@ bool Layer::saveWeights(const string &prjname)
std::cout << "Exporting weights for " << m_name << "." << to_string((int)m_id) << std::endl; 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; int i, j;
for (i=0; i < numVisible; i++) for (i=0; i < numVisible; i++)