MNIST DatasetをChainerで,GPUを利用して学習するまで - automatikもう少ししたら発表があることに気づいて必死でモデルを組み始めた。 ちょっとChainerを思い出しながら書いて,GPUも手に入って環境も整ったこともあり,一回MNISTを分類してみようかと。 いろんなブログやらqiitaを参考にしながらもう一度書いてみた。 CPUモードからGPUモードに変更するときは # GPUの設定 cuda.get_device(0).use() model.to_gpu() # modelの設定後(この場合インスタンス化のあと)に行うこと。 これと xp = cuda.cupy これでnumpyの配列をすべてxpで変えていく。 注意として,numpyにはあったメソッドがxpでなかったりするので,そこらへんはちょろちょろ自分で書いたりしなきゃいけないかも. 今回少しだけはまった点としては def forward(self, x_data, y_data, train=
