Chainerで学習済みモデルを使う方法について

1,179 views

Skip to first unread message

unread,

Jul 21, 2015, 6:50:34 AM7/21/15

to cha...@googlegroups.com

こんにちは、

Chainerの初心者です。使いやすくて、Caffeより楽や早いので助かります。

ありがとうございます。

教えていただきたいことがあります。

今日詰まっていたのは、chainerの各layerの出力を可視化する処理です。その前に、chainerで学習済みのモデルをpickleでダンプして、もう一回loadしました。学習済みモデルはgpu arrayなので、使うときはcuda.initを呼びました。

さらに、

x = chainer.Variable(x.reshape(1, 1, 32, 32), volatile=True)

h = F.relu(F.max_pooling_2d(model.conv1(x), 3, stride=2))

というようなコードを書きましたが、ただプログラムは何もエラーが表示されなく、ずっとここでとまっていただけです。

そもそも、chainerの学習済みモデル（caffe reference modelではなく、chainerで作成されたもの）を使う方法を教えていただけないでしょうか。

よろしくお願いいたします。

unread,

Jul 27, 2015, 2:24:27 AM7/27/15

to Chainer User Group, k09tr...@gmail.com, k09tr...@gmail.com

こんにちは

> Chainerの初心者です。使いやすくて、Caffeより楽や早いので助かります。

ありがとうございます。是非今後も使ってみてください。

> さらに、

> x = chainer.Variable(x.reshape(1, 1, 32, 32), volatile=True)

> h = F.relu(F.max_pooling_2d(model.conv1(x), 3, stride=2))

> というようなコードを書きましたが、ただプログラムは何もエラーが表示されなく、ずっとここでとまっていただけです。

Functionのmodel.conv1やVariableのxはcuda.to_gpuをしてGPU側のメモリで重みや値を確保していますでしょうか。

もしそれでうまくいかないようでしたら、さらに詳しい状況を教えていただけますでしょうか

エラーが発生したコードか、同じ状況が再現するコードを見せていただけると都合が良いです。