pythonのchainerを利用して生成モデルを読み込みたい

Question

前提・実現したいこと

タイトルにあるように、pythonのchainerを利用して生成モデルを読み込みたいです。

chainer.serializers.load_npz()で読み込めると参考サイト等に書かれているのですが、エラーが発生します。
調べてみたのですが原因がわからず、こちらで質問させていただきました。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "bin/eval.py", line 39, in <module>
    serializers.load_npz(args.model_path, gen)
  File "C:\Users****\lib\site-packages\chainer\serializers\npz.py", line 179, in load_npz
    d.load(obj)
  File "C:\Users****\lib\site-packages\chainer\serializer.py", line 83, in load
    obj.serialize(self)
  File "C:\Users****\lib\site-packages\chainer\link.py", line 942, in serialize
    d[name].serialize(serializer[name])
  File "C:\Users****\lib\site-packages\chainer\link.py", line 600, in serialize
    data = serializer(name, param.data)
  File "C:\Users****\lib\site-packages\chainer\serializers\npz.py", line 142, in __call__
    dataset = self.npz[key]
  File "C:\Users****\lib\site-packages\numpy\lib\npyio.py", line 266, in __getitem__
    raise KeyError("%s is not a file in the archive" % key)
KeyError: 'bn3/gamma is not a file in the archive'

該当のソースコード

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('model_path', type=str,
                        help='Generatorの重みファイルへのパス')
    parser.add_argument('--gpu', '-g', type=int, default=0)
    parser.add_argument('--batchsize', '-b', type=int, default=200)
    args = parser.parse_args()

-------------------------------略----------------------------------

gen = projection_gan.pose.posenet.MLP(mode='generator',
        use_bn=opts['use_bn'], activate_func=getattr(F, opts['activate_func']))
    serializers.load_npz(args.model_path, gen)

試したこと

モデルへのパスはしっかり設定できていたと思います。

Answer 1

Chainerのモデルのセーブとロード
上記サイトにあるように、
セーブした時に使ったモジュールをロードした時に使っていないのが原因ではないでしょうか？

#モデルセーブ時にこのように書いたいたはずです。
model = モジュール(myDNN(100, 500, 10))

#しかし、ロードする時にモジュール名を省いたのではないでしょうか？
from chainer import serializers

model = myDNN(100, 500, 10)

#エラー発生
KeyError: '??? is not a file in the archive'

エラーを回避するためには、ロード時にも同じモジュール名を書くと良いのではないでしょうか？

Comments

[maru-ichi]

回答ありがとうございます。

今回、外部サイトよりモデルをダウンロードしているため、セーブ時のモジュールがわかりません。

そのため断定できませんが、ほかのファイルでは同様のソースコードで呼び出すことができているため、モジュール名の欠落とも考えにくい気がします、

何か他に思いつく原因ありましたら教えていただきたいです。

[maru-ichi]

他のファイルと呼び出し形式をそろえてみたところ動作しました。
原因ははっきりしませんが、ご指摘の通り、セーブ時とロード時の差だと思います。

ありがとうございました。