【发布时间】:2018-11-03 11:52:57
【问题描述】:
我在 Digits 上训练了预训练的 GoogleNet(caffe)。培训进展顺利,Digits UI 中的测试显示没有任何问题。但是当我导出模型并尝试使用 example.py 提供的数字时。出现了这个错误。
无法从层“conv1/7x7_s2/bn”复制参数 0 权重;形状错配 H。源参数形状为 1 64 1 1 (64);目标参数形状为 64 (64)。从 sc 学习这一层的参数 ratch 而不是从保存的网络中复制,重命名图层。
我已经检查了 deploy.prototxt 和 train_val.prototxt。尺寸正确。
有什么建议吗?
【问题讨论】:
标签: machine-learning neural-network deep-learning caffe nvidia-digits