diff options
Diffstat (limited to 'src/caffe/layers/hdf5_output_layer.cpp')
| -rw-r--r-- | src/caffe/layers/hdf5_output_layer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/caffe/layers/hdf5_output_layer.cpp b/src/caffe/layers/hdf5_output_layer.cpp index 4a72a18a..d2fdeffc 100644 --- a/src/caffe/layers/hdf5_output_layer.cpp +++ b/src/caffe/layers/hdf5_output_layer.cpp @@ -70,5 +70,6 @@ STUB_GPU(HDF5OutputLayer); #endif INSTANTIATE_CLASS(HDF5OutputLayer); -REGISTER_LAYER_CLASS(HDF5_OUTPUT, HDF5OutputLayer); +REGISTER_LAYER_CLASS(HDF5Output); + } // namespace caffe |