我需要在Chainer框架中加载一个预训练模型,但我了解到,保存的(.npz)文件只包含权重,我必须先重建模型,然后将权重加载到模型中,无法像Tensorflow那样通过一个命令加载整个模型。
这是真的吗?如果是,有使用Chainer框架经验的人能提供一些指导吗?如果不是,那么在上述框架中加载预训练模型的正确方法是什么?
回答:
是的,npz文件只包含权重。你需要先构造模型的一个实例(chainer.Chain
的子类),然后使用load_npz
加载权重。 https://docs.chainer.org/en/stable/guides/serializers.html