update use of GetLayer -> LayerRegistry::CreateLayer
authorJonathan L Long <jonlong@cs.berkeley.edu>
Sat, 27 Dec 2014 09:47:59 +0000 (01:47 -0800)
committerJonathan L Long <jonlong@cs.berkeley.edu>
Sat, 27 Dec 2014 09:47:59 +0000 (01:47 -0800)
src/caffe/net.cpp

index 21ab15f..e4492cf 100644 (file)
@@ -62,7 +62,8 @@ void Net<Dtype>::Init(const NetParameter& in_param) {
   bottom_need_backward_.resize(param.layers_size());
   for (int layer_id = 0; layer_id < param.layers_size(); ++layer_id) {
     const LayerParameter& layer_param = param.layers(layer_id);
-    layers_.push_back(shared_ptr<Layer<Dtype> >(GetLayer<Dtype>(layer_param)));
+    layers_.push_back(shared_ptr<Layer<Dtype> >(
+          LayerRegistry<Dtype>::CreateLayer(layer_param)));
     layer_names_.push_back(layer_param.name());
     LOG(INFO) << "Creating Layer " << layer_param.name();
     bool need_backward = false;