Change TestSnapshot to expect a .h5 snapshot

BVLC · Jul 30, 2015 · 86cdba4 · 86cdba4
1 parent 7e4f178
commit 86cdba4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/test/test_gradient_based_solver.cpp b/src/caffe/test/test_gradient_based_solver.cpp
@@ -139,7 +139,7 @@ class GradientBasedSolverTest : public MultiDeviceTest<TypeParam> {
  if (snapshot) {
  ostringstream resume_file;
  resume_file << snapshot_prefix_ << "/_iter_" << num_iters
- << ".solverstate";
+ << ".solverstate.h5";
  string resume_filename = resume_file.str();
  return resume_filename;
  }