From f75e2dbf7af8875321cd4c05facad4224a0fdf3f Mon Sep 17 00:00:00 2001 From: Davis King Date: Sun, 10 Apr 2016 13:41:05 -0400 Subject: [PATCH] Minor code cleanup --- dlib/dnn/trainer.h | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/dlib/dnn/trainer.h b/dlib/dnn/trainer.h index 2fc1d3ce2..36c32db65 100644 --- a/dlib/dnn/trainer.h +++ b/dlib/dnn/trainer.h @@ -315,7 +315,7 @@ namespace dlib time_between_syncs = time_between_syncs_; // 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) deserialize(*this, fin); } @@ -571,18 +571,17 @@ namespace dlib if (std::chrono::system_clock::now() - last_sync_time > time_between_syncs || 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(); + + // compact network before saving to disk. 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 // file and move the .tmp file to it. - std::remove(sync_filename.c_str()); std::rename(tempfile.c_str(), sync_filename.c_str()); last_sync_time = std::chrono::system_clock::now();