有没有一种简单的方法(例如,不修改 caffe 代码)将多个预训练网络中的 wights 加载到一个网络中?该网络包含一些与预训练网络具有相同维度和名称的层。
我正在尝试使用 NVidia DIGITS 和 Caffe 来实现这一点。
编辑 :正如答案所证实的那样,我认为不可能直接从 DIGITS 中做到这一点。任何人都可以建议一种简单的方法来修改 DIGITS 代码以便能够选择多个预训练网络吗?我查了一下代码,想到了 training script将是一个很好的起点,但我对 Caffe 没有深入的了解,所以我不确定实现这一目标的最佳/最快方法是什么。
最佳答案
正如 Shai 建议的那样,没有办法做到这一点,所以我决定克隆官方存储库并进行适当的更改。我更改了代码,以便可以使用冒号作为分隔符来加载多个预训练网络。
我在官方存储库上创建了一个拉取请求,然后我的更改与 DIGITS 的主分支合并,这意味着现在可以在 DIGITS 中使用此功能。
关于deep-learning - Caffe 中的多个预训练网络,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34466901/