Snapshot model weights/solver state to HDF5 files.
authorEric Tzeng <etzeng@eecs.berkeley.edu>
Wed, 22 Jul 2015 23:17:01 +0000 (16:17 -0700)
committerEric Tzeng <etzeng@eecs.berkeley.edu>
Fri, 7 Aug 2015 21:56:38 +0000 (14:56 -0700)
commitada055bbf155882534907a7fb98a35e4f7bba392
treec91d108a3cab4709c1bf75ece74248a41a5ad3b1
parent1e740e1e8568063ffe69162a30a97961c5c62af0
Snapshot model weights/solver state to HDF5 files.

Summary of changes:
- HDF5 helper functions were moved into a separate file util/hdf5.cpp
- hdf5_save_nd_dataset now saves n-d blobs, can save diffs instead of
  data
- Minor fix for memory leak in HDF5 functions (delete instead of
  delete[])
- Extra methods have been added to both Net/Solver enabling
  snapshotting and restoring from HDF5 files
- snapshot_format was added to SolverParameters, with possible values
  HDF5 or BINARYPROTO (default HDF5)
- kMaxBlobAxes was reduced to 32 to match the limitations of HDF5
14 files changed:
include/caffe/blob.hpp
include/caffe/net.hpp
include/caffe/solver.hpp
include/caffe/util/hdf5.hpp [new file with mode: 0644]
include/caffe/util/io.hpp
src/caffe/layers/hdf5_data_layer.cpp
src/caffe/layers/hdf5_output_layer.cpp
src/caffe/layers/hdf5_output_layer.cu
src/caffe/net.cpp
src/caffe/proto/caffe.proto
src/caffe/solver.cpp
src/caffe/test/test_hdf5_output_layer.cpp
src/caffe/util/hdf5.cpp [new file with mode: 0644]
src/caffe/util/io.cpp