Merge pull request #5215 from cypof/fix_restore

Restore can be invoked on rank > 0
BVLC · Jan 23, 2017 · 51f2986 · 51f2986
2 parents 3babd4e + e0cd852
commit 51f2986
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
@@ -462,7 +462,6 @@ string Solver<Dtype>::SnapshotToHDF5() {
 
 template <typename Dtype>
 void Solver<Dtype>::Restore(const char* state_file) {
-  CHECK(Caffe::root_solver());
   string state_filename(state_file);
   if (state_filename.size() >= 3 &&
       state_filename.compare(state_filename.size() - 3, 3, ".h5") == 0) {