在Chainer深度学习框架中加载预训练的模型

时间:2020-10-17 16:47:53

标签: python tensorflow machine-learning deep-learning chainer

我需要在 Chainer框架中加载预先训练的模型,但是据我了解,保存的(.npz)文件仅包含权重,我必须重建模型然后将权重加载到其中,就无法通过Tensorflow这样的命令来加载整个模型。

这是真的吗?我是这样,任何具有Chainer框架经验的人都可以提供一些指导?如果没有,那么在提到的框架中加载预训练模型的正确方法是什么?

1 个答案:

答案 0 :(得分:1)

是的,只有npz文件仅包含权重。您需要首先构造模型的实例(chainer.Chain的子类),然后使用load_npz在模型上加载权重。 https://docs.chainer.org/en/stable/guides/serializers.html