зеркало из https://github.com/microsoft/caffe.git
Add HDF5OutputLayer to the layer factory
This commit is contained in:
Родитель
e2beba9b55
Коммит
dd9e05bff4
|
@ -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") {
|
||||
|
|
Загрузка…
Ссылка в новой задаче