digits 4.0 0.14.0-rc.3 /Ubuntu (aws)
训练一个 5 类 GoogLenet 模型,每类大约有 800 个训练样本。我试图使用 bvlc_imagent 作为预训练模型。这些是我采取的步骤:
从http://dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemodel下载 imagenet并将其放在 /home/ubuntu/models
2.
一个。将“train_val.prototxt”从此处https://github.com/BVLC/caffe/blob/master/models/bvlc_reference_caffenet/train_val.prototxt粘贴到自定义网络选项卡中,然后
湾。'#' 注释掉了“源”和“后端”行(因为它正在抱怨它们)
- 在预训练模型文本框中粘贴了“.caffemodel”的路径。就我而言:“/home/ubuntu/models/bvlc_googlenet.caffemodel”
我收到此错误:
ERROR: Cannot copy param 0 weights from layer 'loss1/classifier'; shape mismatch. Source param shape is 1 1 1000 1024 (1024000); target param shape is 6 1024 (6144). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.
我已经从 github 问题等中粘贴了各种 train_val.prototext 不幸的是,没有运气,
我不确定为什么这会变得如此复杂,在旧版本的数字中,我们可以输入文件夹的路径,它对于迁移学习非常有用。
有人可以帮忙吗?