Keras好きがPyTorchに挑戦してみた - Qiita

はじめに ずっとKerasでばかりDeep Learningのプログラムを書いてきたが、いよいよPyTorchで書かなくてはいけない状況になってきたので、お勉強がてら、MNISTを例にとりあえず動くものを作ってみました。 【修正履歴】 ・モデル内でFlatten()とSoftmax()を使うようにしました ・対応して、ロス関数をCrossEntropyLoss()にしました まずはチュートリアル とりあえず最初はチュートリアルだろうということで、PyTorchのチュートリアルを最初から始めてみました。 しかし、これが実に分かりづらい。 「Autograd」が感覚的につかめず、モチベーションが10%以下まで落ち込みました。 今では、最初からではなく、「What is torch.nn really?」から初めて、わからないところを調べたほうがよかったと思っています。 バックアップ そんな予感