mxnet ValueError: Parameter *** because it has not been in

导读

在使用mxnet的gluon训练网络的时候，报ValueError: Cannot reset context for Parameter 'mobilenet0_dense1_weight' because it has not been initialized.

错误原因

通过错误信息不难看出是因为参数mobilenet0_dense1_weight没有初始化导致的，所以我们只需要初始化这个参数即可，mxnet有多种可以初始化参数的方法，这里我们介绍两种

• collect_params函数初始化

mxnet提供了一个collect_params方法，可以获取到网络中所有的参数信息，而且该方法还支持正则表达式，比如我们想要初始化mobilenet0_dense1_weight参数，可以这样写

#ctx表示参数是初始化在GPU还是CPU上
net.collect_params("mobilenet0_dense.*_weight").initialize(init.Xavier(),ctx=ctx)

• 对所有为None的参数进行初始化

    for param in finetune_net.collect_params().values():
        if param._data is None:
            print(param)#输出参数信息
            param.initialize()