diff options
-rw-r--r-- | include/caffe/data_layers.hpp | 3 | ||||
-rw-r--r-- | include/caffe/neuron_layers.hpp | 3 |
2 files changed, 3 insertions, 3 deletions
diff --git a/include/caffe/data_layers.hpp b/include/caffe/data_layers.hpp index 90fd0d19..aa0ab7df 100644 --- a/include/caffe/data_layers.hpp +++ b/include/caffe/data_layers.hpp @@ -17,6 +17,9 @@ #include "caffe/util/blocking_queue.hpp" #include "caffe/util/db.hpp" +#define HDF5_DATA_DATASET_NAME "data" +#define HDF5_DATA_LABEL_NAME "label" + namespace caffe { /** diff --git a/include/caffe/neuron_layers.hpp b/include/caffe/neuron_layers.hpp index c2e0774a..4fa330ec 100644 --- a/include/caffe/neuron_layers.hpp +++ b/include/caffe/neuron_layers.hpp @@ -10,9 +10,6 @@ #include "caffe/layer.hpp" #include "caffe/proto/caffe.pb.h" -#define HDF5_DATA_DATASET_NAME "data" -#define HDF5_DATA_LABEL_NAME "label" - namespace caffe { /** |