singa-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From wang...@apache.org
Subject incubator-singa git commit: SINGA-12 Supprt Checkpoint and Restore
Date Tue, 28 Jul 2015 11:59:29 GMT
Repository: incubator-singa
Updated Branches:
  refs/heads/master 4c289744d -> 06163950b


SINGA-12 Supprt Checkpoint and Restore

Fixbug from Resume function, which generated errors when irregular files are put into the
checkpoint folder.
Now irregular files will be reported and ignored.


Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/06163950
Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/06163950
Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/06163950

Branch: refs/heads/master
Commit: 06163950bff355ce3c83764ab51f07ee95993e09
Parents: 4c28974
Author: Wei Wang <wangwei@comp.nus.edu.sg>
Authored: Tue Jul 28 19:50:10 2015 +0800
Committer: Wei Wang <wangwei@comp.nus.edu.sg>
Committed: Tue Jul 28 19:50:10 2015 +0800

----------------------------------------------------------------------
 src/trainer/trainer.cc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/06163950/src/trainer/trainer.cc
----------------------------------------------------------------------
diff --git a/src/trainer/trainer.cc b/src/trainer/trainer.cc
index d5c885f..9f245a2 100644
--- a/src/trainer/trainer.cc
+++ b/src/trainer/trainer.cc
@@ -207,8 +207,9 @@ void Trainer::Resume(ModelProto* modelConf) {
     tinydir_readfile(&dir, &file);
     tinydir_next(&dir);
     char* ch = strstr(file.name, "step");
-    if (ch == nullptr && file.name[0] != '.' && file.name[1] != '.') {
-      LOG(INFO) << "Irregular file in checkpoint folder: " << file.name;
+    if (ch == nullptr) {
+      if (file.name[0] != '.')
+        LOG(INFO) << "Irregular file in checkpoint folder: " << file.name;
       continue;
     }
 


Mime
View raw message