diff options
Diffstat (limited to 'src/caffe/layers/hdf5_output_layer.cu')
-rw-r--r-- | src/caffe/layers/hdf5_output_layer.cu | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/src/caffe/layers/hdf5_output_layer.cu b/src/caffe/layers/hdf5_output_layer.cu index eb6d0e47..ca8f2616 100644 --- a/src/caffe/layers/hdf5_output_layer.cu +++ b/src/caffe/layers/hdf5_output_layer.cu @@ -3,10 +3,7 @@ #include "hdf5.h" #include "hdf5_hl.h" -#include "caffe/blob.hpp" -#include "caffe/common.hpp" -#include "caffe/layer.hpp" -#include "caffe/vision_layers.hpp" +#include "caffe/data_layers.hpp" namespace caffe { |