summaryrefslogtreecommitdiff
path: root/src/caffe/solver.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/caffe/solver.cpp')
-rw-r--r--src/caffe/solver.cpp16
1 files changed, 8 insertions, 8 deletions
diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
index 3574ce7..12c13dd 100644
--- a/src/caffe/solver.cpp
+++ b/src/caffe/solver.cpp
@@ -422,14 +422,14 @@ void Solver<Dtype>::Snapshot() {
CHECK(Caffe::root_solver());
string model_filename;
switch (param_.snapshot_format()) {
- case caffe::SolverParameter_SnapshotFormat_BINARYPROTO:
- model_filename = SnapshotToBinaryProto();
- break;
- case caffe::SolverParameter_SnapshotFormat_HDF5:
- model_filename = SnapshotToHDF5();
- break;
- default:
- LOG(FATAL) << "Unsupported snapshot format.";
+ case caffe::SolverParameter_SnapshotFormat_BINARYPROTO:
+ model_filename = SnapshotToBinaryProto();
+ break;
+ case caffe::SolverParameter_SnapshotFormat_HDF5:
+ model_filename = SnapshotToHDF5();
+ break;
+ default:
+ LOG(FATAL) << "Unsupported snapshot format.";
}
SnapshotSolverState(model_filename);