我已经在 theano/lasagne 中训练了一个神经网络,并用这段代码保存了网络参数:
np.savez('model.npz', *lasagne.layers.get_all_param_values(network))
现在我想加载模型并将参数设置为网络,以便测试它是否正常工作。我正在使用这段代码加载参数:
with np.load('model.npz') as f:
param_values = [f['arr_%d' % i] for i in range(len(f.files))]
然后我尝试将网络的参数设置为加载的参数,如下所示:
lasagne.layers.set_all_param_values(self._output_layer, param_values, allow_input_downcast=True)
但后来我得到这个错误:
> Traceback (most recent call last):
File "snake_player.py", line 406, in <module>
player = SnakePlayer()
File "snake_player.py", line 99, in __init__
self.loadModel()
File "snake_player.py", line 328, in loadModel
lasagne.layers.set_all_param_values(self._output_layer, param_values, allow_input_downcast=True, trainable=True)
File "C:\Anaconda2\lib\site-packages\lasagne\layers\helper.py", line 446, in set_all_param_values
(len(values), len(params)))
ValueError: mismatch: got 10 values to set 0 parameters
然而,当我得到网络中的所有参数并查看长度时,有 10 个参数。
x = lasagne.layers.get_all_param_values(self._output_layer)
len(x) # 10
所以参数的数量是相同的,不知何故我仍然得到这个错误。知道为什么吗?我忽略了什么吗?
我很感激帮助。