是否有一种简单的方法(例如,不修改caffe代码)将多个预训练网络的重量加载到一个网络中?该网络包含一些与预训练网络具有相同尺寸和名称的图层.
我试图用NVidia DIGITS和Caffe实现这一目标.
编辑:我认为不可能直接从DIGITS这样做,正如答案所证实的那样.任何人都可以建议一种简单的方法来修改DIGITS代码,以便能够选择多个预训练网络吗?我检查了一下代码,并认为训练脚本是一个很好的起点,但我对Caffe没有深入的了解,所以我不确定实现这个目标的最佳/最快方法是什么.
deep-learning caffe conv-neural-network pycaffe nvidia-digits
包 Digits 需要使用 Caffe 安装目录的位置设置环境变量。
安装 Caffe 的简单方法是apt-get install caffe-cuda
. 但是,我无法弄清楚它的安装位置。它没有安装在主目录中,dpkg -L caffe-cuda
只显示其中的几个文件/usr/share/doc
.
有任何想法吗?