From f439f177297d35fa2998f2d43e2a278dd18fb96d Mon Sep 17 00:00:00 2001 From: SunAhong1993 <48579383+SunAhong1993@users.noreply.github.com> Date: Fri, 9 Aug 2019 13:00:19 +0800 Subject: [PATCH] Update add_caffe_custom_layer.md --- add_caffe_custom_layer.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/add_caffe_custom_layer.md b/add_caffe_custom_layer.md index af5827b..c308135 100644 --- a/add_caffe_custom_layer.md +++ b/add_caffe_custom_layer.md @@ -32,7 +32,7 @@ bash ./toos/compile.sh /home/root/caffe/src/caffe/proto | name | str(默认为None) | mylayer的名字 | | 其余 | 默认为None | 命名为Caffe模型的model.prototxt中mylayer_param中每个参数的名字 | - 功能:运用PaddlePaddle完成组网来实现`mylayer`的功能 + 功能:运用PaddlePaddle完成组网来实现`mylayer`的功能 返回:一个Variable或Tensor,为组网后的结果 3. `def mylayer_weights(name, data=None)` @@ -42,7 +42,7 @@ bash ./toos/compile.sh /home/root/caffe/src/caffe/proto | name | str | mylayer的名字 | | data | list(默认为None) | 由Caffe模型的model.caffemodel获得的关于mylayer的参数 | - 功能:为每个参数(例如kernel、bias等)命名;同时,若Caffe中该层参数与PaddlePaddle中参数的格式不一致,则需要的变换操作也在该函数中实现。 + 功能:为每个参数(例如kernel、bias等)命名;同时,若Caffe中该层参数与PaddlePaddle中参数的格式不一致,则需要的变换操作也在该函数中实现。 返回:一个list,包含每个参数的名字。 - 在mylayer.py中注册`mylayer`,主要运用下述代码实现: -- GitLab