caffe2获得学习率:找不到blob: gpu_0/卷积_w_lr

Question

在resnet50_train.py中获取学习率的代码如下：

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

当我运行代码时，会发生以下错误：

追溯(最近一次调用)：文件"/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py"，行475，主()文件"/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py"，行471，主列车文件"/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py"，行400，在列车爆炸文件"/home/caffe2/caffe2/caffe2/python/examples/resnet50_trainer.py"，第163行中，在RunEpoch learning_rate =workspace.FetchBlob(前缀+ '/conv1_w_lr')文件"/home/caffe2-master/caffe2/build/caffe2/python/workspace.py"，行323中，在FetchBlob返回C.fetch_blob(StringifyBlobName(名称))RuntimeError中:在pybind_state.cc:152 ws->HasBlob(name)中强制执行失败。找不到blob: gpu_0/卷积_w_lr

是什么引起了这个问题？我是否应该重新编译任何依赖项，或者如果有任何其他功能可以用来获得学习率？

Answer 1

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

也许conv1_w_lr没有前缀而存在。也许，您首先需要RunNet才能使工作区中的所有blobs。