Question

关于如何微调参数的信息非常少，而且我对如何微调caffe2中的网络真的很困惑。任何人都可以给我看一些关于微调部分的代码。非常感谢。

顺便说一句，在链接：Food101 SqueezeNet Caffe2 number of iterations中，似乎作者已经成功地对网络进行了微调。

add：以下是我的火车部分的一些代码，

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

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

如何在caffe2中微调网络

0 个答案: