/caffe

Restore can be invoked on rank > 0 #5215

Merged
merged 1 commit into from Jan 23, 2017
Commits
Jump to file or symbol
Failed to load files and symbols.
+0 −1
Split
View
1 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) {