質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.50%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Q&A

解決済

1回答

1024閲覧

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

maru-ichi

総合スコア8

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

0グッド

0クリップ

投稿2019/12/09 10:38

前提・実現したいこと

タイトルにあるように、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'

該当のソースコード

python3

1if __name__ == '__main__': 2 parser = argparse.ArgumentParser() 3 parser.add_argument('model_path', type=str, 4 help='Generatorの重みファイルへのパス') 5 parser.add_argument('--gpu', '-g', type=int, default=0) 6 parser.add_argument('--batchsize', '-b', type=int, default=200) 7 args = parser.parse_args() 8 9-------------------------------略---------------------------------- 10 11gen = projection_gan.pose.posenet.MLP(mode='generator', 12 use_bn=opts['use_bn'], activate_func=getattr(F, opts['activate_func'])) 13 serializers.load_npz(args.model_path, gen)

試したこと

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

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

ベストアンサー

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

python

1#モデルセーブ時にこのように書いたいたはずです。 2model = モジュール(myDNN(100, 500, 10)) 3 4#しかし、ロードする時にモジュール名を省いたのではないでしょうか? 5from chainer import serializers 6 7model = myDNN(100, 500, 10) 8 9#エラー発生 10KeyError: '??? is not a file in the archive'

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

投稿2019/12/09 10:58

yep.app

総合スコア11

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

maru-ichi

2019/12/10 21:53

回答ありがとうございます。 今回、外部サイトよりモデルをダウンロードしているため、セーブ時のモジュールがわかりません。 そのため断定できませんが、ほかのファイルでは同様のソースコードで呼び出すことができているため、モジュール名の欠落とも考えにくい気がします、 何か他に思いつく原因ありましたら教えていただきたいです。
maru-ichi

2019/12/10 22:38

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

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.50%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問