OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Made trainer disk synchronization more reliable and efficient.

This commit is contained in:
Davis King 2016-01-09 11:57:04 -05:00
parent 08f965a32b
commit 4b2178c6e6
1 changed files with 5 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dlib/dnn/trainer.h
View File

@ -281,21 +281,11 @@ namespace dlib
sync_filename = filename;
time_between_syncs = time_between_syncs_;
// check if the sync file already exists, if it does we should load it. We
// first check for a .tmp version since that would be the newest if it existed.
// If it doesn't exist we check the canonical sync file.
std::ifstream fin(filename+".tmp", std::ios::binary);
if (fin)
{
deserialize(*this, fin);
}
else
{
// check if the sync file already exists, if it does we should load it.
std::ifstream fin(filename, std::ios::binary);
if (fin)
deserialize(*this, fin);
}
}
double get_average_loss (
) const
@ -521,6 +511,9 @@ namespace dlib
// save our state to a temp file
std::string tempfile = sync_filename + ".tmp";
std::ofstream fout(tempfile, std::ios::binary);
// compact network before saving to disk.
wait_for_thread_to_pause();
this->net.clean();
serialize(*this, fout);
fout.close();