概要

研究において、GPUで学習してもらったPickleのdumpファイルはGPUで作ったPickleのファイルであるため、どうやらCPUでそのまま使えないらしい。実際にやってみたところエラーが出たので改善策をまとめておく。

このままではせっかく作ったモデルファイルで予測ができない。
よってGPUで作ったpickleのモデルファイルを使って予測を行うために自分がやったことをメモする。

環境

pylearn2にあるgpu_pkl_to_cpu_pklを使ってdumpファイルをcpu用に変換する。
ただし、これはどうやら実験段階にあるため、環境によってはそもそもうまくいかない場合があることに注意する。

[画像]

変換後、予測のタスクをやってみたところしっかり予測できた。
エラーもなし。スバラシイ。

そもそもGPUで保存する際にCPUで使えるように保存できないのかと思って調べてみたところ

model.to_cpu
pickle.dump()

これでchainerの場合はCPU向けのPickleにできるらしい。センセイェ...
※GPUでの学習において、このpickle.dumpをepochごとに行なっている場合は、次のepochを始める際にmodel.to_gpuをしておかないと、モデルがおかしくなるので注意が必要。