Change TestSnapshot to expect a .h5 snapshot

erictzeng · erictzeng · commit 86cdba40bf5c · 2015-07-29T22:40:16.000-07:00
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;
     }

Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,7 @@ class GradientBasedSolverTest : public MultiDeviceTest<TypeParam> {`
`139`	`139`	`if (snapshot) {`
`140`	`140`	`ostringstream resume_file;`
`141`	`141`	`resume_file << snapshot_prefix_ << "/_iter_" << num_iters`
`142`		`- << ".solverstate";`
	`142`	`+ << ".solverstate.h5";`
`143`	`143`	`string resume_filename = resume_file.str();`
`144`	`144`	`return resume_filename;`
`145`	`145`	`}`