Python caffe2获得学习率：Can'；找不到blob:gpu_0/conv1_w_lr

在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/python/examples/resnet50_trainer.py”，第行 总的来说是471 训练（args）文件“/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py”，第行 400，在火车上 explog文件“/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py”，第行 163，在新纪元 learning_rate=workspace.FetchBlob（前缀+'/conv1_w_lr'）文件“/home/caffe2 master/caffe2/build/caffe2/python/workspace.py”，第行 323，在FetchBlob中 返回C.fetch_blob（StringifyBlobName（name））运行时错误：[在pybind_状态下强制失败。cc:152]ws->HasBlob（name）。找不到 blob:gpu_0/conv1_w_lr

---

是什么导致了这个问题？我是否应该重新编译任何依赖项，或者是否可以使用任何其他函数来获取学习率？

在运行

FetchBlob（前缀+'/conv1\u w\u lr'）

之前，您可以检查工作区中存在哪些blob:

工作区中的b。blob（）：打印（b）

---

可能存在没有前缀的

conv1\u w\u lr

。也许，您首先需要运行RunNet，以便在工作区中保存所有blob。

非常感谢。我在github上查看了resnet50_trainer.py，发现它与我下载的版本不同。获得学习率的方式已经改变。。。learning\u rate=workspace.FetchBlob（数据\u并行\u模型.GetLearningRateBlobNames（训练模型）[0]）重新编译了最新的caffe2，现在一切正常。