gpt4 book ai didi

machine-learning - 在 Caffe 中创建新层

转载 作者:行者123 更新时间:2023-11-30 08:33:11 24 4
gpt4 key购买 nike

我使用caffe为了我的深度学习。要为 caffe 创建一个新层,我需要

(1)向proto/caffe.proto添加层以更新下一个可用的ID

(2)通过将其添加到layer_factory.cpp使图层可创建

我的疑问是caffe.proto中新ID的用途是什么以及layer_factory.cpp中createable的功能。

有些层未包含在layer_factory.cpp 中作为可创建层。 layer_factory.cpp 中包含和未包含的那些层有什么区别。

最佳答案

如果你想在caffe中添加新的层,请按照wiki/Development中的说明进行操作。 .
您会看到不需要更改 layer_factory.cpp,而是在 cpp 文件中添加两个宏 INSTANTIATE_CLASSREGISTER_LAYER_CLASS

关于caffe.proto:如果您希望您的图层具有特定的my_layer_param消息,则只需在此处进行更改。如果您的图层重复使用现有图层的参数,则根本不需要更改 caffe.proto

关于machine-learning - 在 Caffe 中创建新层,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45649515/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com