| 28 | |
| 29 | template <typename Dtype> |
| 30 | void HDF5OutputLayer<Dtype>::SaveBlobs() { |
| 31 | // TODO: no limit on the number of blobs |
| 32 | LOG(INFO) << "Saving HDF5 file " << file_name_; |
| 33 | CHECK_EQ(data_blob_.num(), label_blob_.num()) << |
| 34 | "data blob and label blob must have the same batch size"; |
| 35 | hdf5_save_nd_dataset(file_id_, HDF5_DATA_DATASET_NAME, data_blob_); |
| 36 | hdf5_save_nd_dataset(file_id_, HDF5_DATA_LABEL_NAME, label_blob_); |
| 37 | LOG(INFO) << "Successfully saved " << data_blob_.num() << " rows"; |
| 38 | } |
| 39 | |
| 40 | template <typename Dtype> |
| 41 | void HDF5OutputLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, |