假设新增加的层命名为:NEW
1. 在src/proto的LayerParameter 的 LayerType下 加 NEW= 数字;
2. 在src/layer_factory.cpp中, 加
case LayerParameter_LayerType_NEW:
return new NewLayer<Dtype>(param);
3.在src/layers/下 加 new_layer.cpp和 new_layer.cu代码;
4.在include/caffe/vision_layers.hpp 下增加代码(也可能在其他的.hpp下增加,如 common_layer.hpp,neuron_layer.hpp等,具体视增加的layer类型决定);5.在upgrade_proto.cpp下增加对应的注册的代码。
caffe中增加自己的layer
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
基于香橙派AIpro将开源框架模型转换为升腾模型
原創
2024-04-12 10:32:18
网络TCP
osc_61miaq6u
2024-05-13 22:58:24
意派Epub360丨518国际博物馆日H5,每个都大有看头!
原創
2024-04-25 22:25:00
意派Epub360丨五四青年节H5竟然可以这young精彩!
原創
2024-04-19 22:24:49
意派Epub360丨探秘车企H5营销策略,线下线上齐发力!
原創
2024-03-28 22:36:28
大模型并行训练指南:Megatron-DeepSpeed的模型并行与数据并行
原創
2024-02-29 00:50:30
简单地聊一聊Spring Boot的构架
原創
2023-11-14 10:55:48
意派Epub360丨线上展览H5,一个字,绝!
原創
2023-11-03 11:08:34
意派Epub360丨中秋H5案例合集,这样的“显眼包”麻烦多来几个!
原創
2023-09-08 22:44:13
Linux 内存管理新特性 - Memory folios 解读
原創
2023-09-08 00:32:42
01-2015-Deep Learning论文翻译精读
原創
2023-09-03 11:12:23
意派Epub360丨【全行业适用】H5小白刚需,手册类H5制作指南
原創
2023-08-03 22:50:42
意派Epub360丨给跪了!这么牛的PC端交互H5案例你不看看?
原創
2023-07-27 22:45:32