studylog/北の雲

chainer/python/nlp

chainerメモ2

苦戦中

CPUでは学習が進むけどGPUだと進まない問題に遭遇。
model.W.dataで重みを見るとbackwordしても変わって無いのであれ?と思ってlossを見たらF.mean_squared_error()が返してるデータがCPUとGPUで何か違う。

#GPUはちゃんとcuda.to_gpuを通した前提で
a = Variable(np.array( [[0,0]] ))
b = Variable(np.array( [[1,1]] ))

loss = F.mean_squared_error(a, b)

loss.data # CPUだと [1.]  GPUだと 0.0が返ってくる

fix: return type of Accuracy#forwad_cpu() norm to array fix#97 by kikusu · Pull Request #100 · pfnet/chainer · GitHub
Behavior of 0-dimensional array · Issue #107 · pfnet/chainer · GitHub

この辺だろうか。わからない。