ssd訓練時提示：Cannot copy param 0 weights from 'xxxx'，以及提示No module named caffe.proto,推理時設置GPU模式

1、Cannot copy param 0 weights from 'xxxx'

使用預訓練的模型，當希望修改部分網絡結構並繼續使用相同結構的網絡參數時，經常遇見這個問題。以爲改的天衣無縫，但是這個問題始終存在，而且無視你的修改。比如，網絡層 conv1明明已經改成了conv_1了，可是還是提示Cannot copy param 0 weights from 'conv1'。。。。後來才發現，訓練的時候也是會用到test.prototxt的，因此必須同時將test裏面的模型也改了，才能順利進行。

記錄一下，並惋惜一下浪費掉的兩天時間。

 

2、 No module named caffe.proto 解決方法

參考下面的鏈接：

https://blog.csdn.net/curious999/article/details/81225624

就是一個.py文件中的絕對路徑出錯了。

3、使用GPU進行部署

caffe提供的SSD的demo默認是CPU模式，在deploy中設置爲CUDNN也沒用，這時候需要在運行前設置一句話，caffe.set_mode_gpu()，這時候你會發現模型推理速度會提升一個數量級！驚喜～～