学习速率：CAFBU2无法获得_

Question

在resnet50中获取学习率的代码_火车.py具体如下：

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

当我运行代码时，出现错误：

Traceback (most recent call last): File "/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py", line 475, in main() File "/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py", line 471, in main Train(args) File "/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py", line 400, in Train explog File "/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py", line 163, in RunEpoch learning_rate = workspace.FetchBlob(prefix + '/conv1_w_lr') File "/home/caffe2-master/caffe2/build/caffe2/python/workspace.py", line 323, in FetchBlob return C.fetch_blob(StringifyBlobName(name)) RuntimeError: [enforce fail at pybind_state.cc:152] ws->HasBlob(name). Can't find blob: gpu_0/conv1_w_lr

问题是什么引起的？我是否应该重新编译任何依赖项，或者是否有任何其他函数可以用来获得学习率？在

Answer 1

在运行FetchBlob(prefix + '/conv1_w_lr')之前，可以检查工作区中存在哪些blob:for b in workspace.Blobs(): print(b)。在

可能conv1_w_lr不带前缀而存在。也许，您首先需要运行net才能在工作区中拥有所有blob。在