未验证 提交 fb71e9dd 编写于 作者: S SunAhong1993 提交者: GitHub

Update caffe_op_mapper.py

上级 7e68465e
...@@ -1120,7 +1120,7 @@ class CaffeOpMapper(OpMapper): ...@@ -1120,7 +1120,7 @@ class CaffeOpMapper(OpMapper):
"pooled_width": params.pooled_w, "pooled_width": params.pooled_w,
"spatial_scale": params.spatial_scale} "spatial_scale": params.spatial_scale}
self.paddle_graph.add_layer( self.paddle_graph.add_layer(
"custom_layer:roipooling", "custom_layer:ROIPooling",
inputs=inputs_dict, inputs=inputs_dict,
outputs=layer_outputs, outputs=layer_outputs,
**layer_attrs) **layer_attrs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册