gpt4 book ai didi

deep-learning - Caffe 中的多个预训练网络

转载 作者:行者123 更新时间:2023-12-04 11:35:49 25 4
gpt4 key购买 nike

有没有一种简单的方法(例如,不修改 代码)将多个预训练网络中的 wights 加载到一个网络中?该网络包含一些与预训练网络具有相同维度和名称的层。

我正在尝试使用 NVidia DIGITS 和 Caffe 来实现这一点。

编辑 :正如答案所证实的那样,我认为不可能直接从 DIGITS 中做到这一点。任何人都可以建议一种简单的方法来修改 DIGITS 代码以便能够选择多个预训练网络吗?我查了一下代码,想到了 training script将是一个很好的起点,但我对 Caffe 没有深入的了解,所以我不确定实现这一目标的最佳/最快方法是什么。

最佳答案

正如 Shai 建议的那样,没有办法做到这一点,所以我决定克隆官方存储库并进行适当的更改。我更改了代码,以便可以使用冒号作为分隔符来加载多个预训练网络。

我在官方存储库上创建了一个拉取请求,然后我的更改与 DIGITS 的主分支合并,这意味着现在可以在 DIGITS 中使用此功能。

关于deep-learning - Caffe 中的多个预训练网络,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34466901/

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