diff options
Diffstat (limited to 'src/caffe/util/io.cpp')
| -rw-r--r-- | src/caffe/util/io.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/util/io.cpp b/src/caffe/util/io.cpp index 6fa6e680..f738e053 100644 --- a/src/caffe/util/io.cpp +++ b/src/caffe/util/io.cpp @@ -256,7 +256,7 @@ void hdf5_load_nd_dataset<double>(hid_t file_id, const char* dataset_name_, template <> void hdf5_save_nd_dataset<float>( - const hid_t file_id, const string dataset_name, const Blob<float>& blob) { + const hid_t file_id, const string& dataset_name, const Blob<float>& blob) { hsize_t dims[HDF5_NUM_DIMS]; dims[0] = blob.num(); dims[1] = blob.channels(); @@ -269,7 +269,7 @@ void hdf5_save_nd_dataset<float>( template <> void hdf5_save_nd_dataset<double>( - const hid_t file_id, const string dataset_name, const Blob<double>& blob) { + const hid_t file_id, const string& dataset_name, const Blob<double>& blob) { hsize_t dims[HDF5_NUM_DIMS]; dims[0] = blob.num(); dims[1] = blob.channels(); |