caffe2獲得學習率：找不到斑點：gpu_0 / conv1_w_lr

Question

在resnet50_train.py中獲得學習率的代碼如下：

learning_rate = workspace.FetchBlob(prefix + '/conv1_w_lr')

當我運行代碼時，發生了錯誤：

追溯（最近一次通話）：文件“ /home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py”，行475，位於main（）文件中，文件“ / home / caffe2 / caffe2 / caffe2 / caffe2 / python / examples /火車主文件（/ args）文件“ /home/caffe2/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py”，行471，在火車explog文件“ / home / caffe2 / caffe2 / caffe2”中，行471 /python/examples/resnet50_trainer.py“，第163行，位於RunEpoch learning_rate =工作區中。FetchBlob（前綴+'/ conv1_w_lr'）文件” /home/caffe2-master/caffe2/build/caffe2/python/workspace.py“，第323行，在FetchBlob中返回C.fetch_blob（StringifyBlobName（name））RuntimeError：[在pybind_state.cc:152上執行失敗] ws-> HasBlob（name）。 找不到Blob：gpu_0 / conv1_w_lr

是什么原因引起的？ 我應該重新編譯任何依賴項，還是可以使用任何其他函數來獲得學習率？

Answer 1

在運行FetchBlob(prefix + '/conv1_w_lr')您可以檢查工作空間中存在哪些blob： for b in workspace.Blobs(): print(b) 。

可能conv1_w_lr存在而沒有前綴。 也許，您首先需要RunNet才能在工作區中包含所有Blob。