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

- Layer: removed legacy weights load/save

Browse Source 
git-svn-id: http://moon:8086/svn/software/trunk/projects/Rbm@777 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
jens
2022-01-10 15:48:51 +00:00
parent bfbd837ed5
commit 03a6624017
3 changed files with 36 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/Stack.cpp
+13 -18
View File
@@ -233,21 +233,19 @@ arma::mat Stack::trainingBatch(Layer* pThatLayer)
size_t Stack::loadTrainingBatch(bool doNormalize)
{
{
std::string path = m_dir + "/" + m_name + ".training.mat";
bool success = m_trainingBatch.load(path, arma::arma_ascii);
std::string filename = m_dir + "/" + m_name + ".training.dat";
std::string path = m_dir + "/" + m_name + ".training.dat";
bool success = m_trainingBatch.load(filename, arma::arma_ascii);
if (success)
{
std::cout << "Loaded " << m_trainingBatch.n_rows << " training samples\n";
return m_trainingBatch.n_rows;
}
if (success)
{
std::cout << "Loaded " << m_trainingBatch.n_rows << " training samples\n";
return m_trainingBatch.n_rows;
}
uint32_t numTraining = 0;
uint32_t numVisible = 0;
std::string filename = m_dir + "/" + m_name + ".training.dat";
FILE *pFile = fopen(filename.c_str(), "r");
if (!pFile)
@@ -293,18 +291,15 @@ size_t Stack::loadTrainingBatch(bool doNormalize)
size_t Stack::saveTrainingBatch()
{
{
std::string path = m_dir + "/" + m_name + ".training.mat";
bool success = m_trainingBatch.save(path, arma::arma_ascii);
std::string filename = m_dir + "/" + m_name + ".training.dat";
bool success = m_trainingBatch.save(filename, arma::arma_ascii);
if (success)
{
std::cout << "Saved " << m_trainingBatch.n_rows << " training samples\n";
return m_trainingBatch.n_rows;
}
if (success)
{
std::cout << "Saved " << m_trainingBatch.n_rows << " training samples\n";
return m_trainingBatch.n_rows;
}
std::string filename = m_dir + "/" + m_name + ".training.dat";
FILE *pFile = fopen(filename.c_str(), "w");
if (!pFile)