Add HDF5OutputLayer to the layer factory

This commit is contained in:
Kai Li 2014-03-23 19:05:05 +08:00
Родитель e2beba9b55
Коммит dd9e05bff4
1 изменённых файлов: 2 добавлений и 0 удалений

Просмотреть файл

@ -37,6 +37,8 @@ Layer<Dtype>* GetLayer(const LayerParameter& param) {
return new FlattenLayer<Dtype>(param);
} else if (type == "hdf5_data") {
return new HDF5DataLayer<Dtype>(param);
} else if (type == "hdf5_output") {
return new HDF5OutputLayer<Dtype>(param);
} else if (type == "images") {
return new ImagesLayer<Dtype>(param);
} else if (type == "im2col") {