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

Minor code cleanup

This commit is contained in:
Davis King 2016-04-10 13:41:05 -04:00
parent 67a81c1c51
commit f75e2dbf7a
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dlib/dnn/trainer.h
View File

@ -315,7 +315,7 @@ namespace dlib
time_between_syncs = time_between_syncs_; time_between_syncs = time_between_syncs_;
// check if the sync file already exists, if it does we should load it. // check if the sync file already exists, if it does we should load it.
std::ifstream fin(filename, std::ios::binary); std::ifstream fin(sync_filename, std::ios::binary);
if (fin) if (fin)
deserialize(*this, fin); deserialize(*this, fin);
} }
@ -571,18 +571,17 @@ namespace dlib
if (std::chrono::system_clock::now() - last_sync_time > time_between_syncs || if (std::chrono::system_clock::now() - last_sync_time > time_between_syncs ||
do_it_now) do_it_now)
{ {
// 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(); wait_for_thread_to_pause();
// compact network before saving to disk.
this->net.clean(); this->net.clean();
serialize(*this, fout);
fout.close(); // save our state to a temp file
const std::string tempfile = sync_filename + ".tmp";
serialize(tempfile) << *this;
// Now that we know the state is safely saved to disk, delete the old sync // Now that we know the state is safely saved to disk, delete the old sync
// file and move the .tmp file to it. // file and move the .tmp file to it.
std::remove(sync_filename.c_str());
std::rename(tempfile.c_str(), sync_filename.c_str()); std::rename(tempfile.c_str(), sync_filename.c_str());
last_sync_time = std::chrono::system_clock::now(); last_sync_time = std::chrono::system_clock::now();