1

关于如何微调参数的信息很少，关于如何在 caffe2 中微调网络真的让我很困惑。谁能给我看一些关于微调部分的代码。非常感谢。

顺便说一句，在链接：Food101 SqueezeNet Caffe2 number of iterations，看来作者已经成功地微调了网络。

补充：这是我的火车部分的一些代码，

train_model = cnn.CNNModelHelper(order="NCHW", name="train")
train_model.param_init_net.AppendNet(core.Net(init_net))
train_model.net.AppendNet(core.Net(predict_net))
train_model.param_init_net.RunAllOnGPU(gpu_id=0)
train_model.net.RunAllOnGPU(gpu_id=0)
workspace.RunNetOnce(train_model.param_init_net)
AddTrainingOperators(train_model, 'softmaxout', 'label')
AddBookkeepingOperators(train_model)
workspace.RunNetOnce(train_model.param_init_net)
data, label = AddInput(train_model, batch_size=3,
                db=os.path.join(data_folder, 'toy_train.lmdb'),
                db_type='lmdb')
workspace.FeedBlob('data', data)
workspace.FeedBlob('label', label)
workspace.CreateNet(train_model.net)

但是，当我运行代码时，出现警告的错误

    Traceback (most recent call last):
  File "lenetForChineseFinetune.py", line 62, in <module>
    workspace.FeedBlob('data', data)
  File "/opt/caffe2/caffe2/local/caffe2/python/workspace.py", line 262, in FeedBlob
    return C.feed_blob(name, arr)
RuntimeError: [enforce fail at pybind_state.cc:825] . Unexpected type of argument - only numpy array or string are supported for feeding

发生。我应该如何修改代码？

4

0 回答 0