Chainerのモデルのセーブとロード - 無限グミChainerのモデルのセーブ・ロードについて Chainerは作成したモデルを書き出すためのserializersを含んでいます。しかし、セーブの方法は多くのページに書いてありますが、ロードの方法がよくわからなかったため自分の環境でうまく行った方法をまとめます。 モデルのセーブ Chainerでモデルをセーブするにはchainer.serialisersを使います。この時にGPU用に変換したものはCPU用に変えておいたほうが後で混乱しないで済むので統一しておくといいと思います。 from chainer import serializers gpu = 0 model = myDNN(100, 500, 10) chainer.cuda.get_device(gpu).use() model.to_gpu(gpu) # GPUを使うための処理 # --------------------
