diff options
| author | Sergio <sguada@gmail.com> | 2014-10-16 11:17:24 -0700 |
|---|---|---|
| committer | Sergio <sguada@gmail.com> | 2014-12-21 17:00:05 -0800 |
| commit | cb3dbee20a46e822c2e4306b03f4b127c1db1f5c (patch) | |
| tree | 9fc70f8db48ff7ebe747819f4e61ad2e61a2a252 /src/caffe | |
| parent | 899dfde6f545faa73c0ffe5732ae641b10dfbc09 (diff) | |
| download | caffeonacl-cb3dbee20a46e822c2e4306b03f4b127c1db1f5c.tar.gz caffeonacl-cb3dbee20a46e822c2e4306b03f4b127c1db1f5c.tar.bz2 caffeonacl-cb3dbee20a46e822c2e4306b03f4b127c1db1f5c.zip | |
Fix init current_step
Diffstat (limited to 'src/caffe')
| -rw-r--r-- | src/caffe/solver.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp index 886c3cc6..0fd8d0f8 100644 --- a/src/caffe/solver.cpp +++ b/src/caffe/solver.cpp @@ -162,6 +162,7 @@ void Solver<Dtype>::Solve(const char* resume_file) { PreSolve(); iter_ = 0; + current_step_ = 0; if (resume_file) { LOG(INFO) << "Restoring previous solver status from " << resume_file; Restore(resume_file); |