p.21の損失収束のグラフに関しての質問 · Issue #6 · smly/kaggle-book-gokui

本のコードの通り学習させてみたのですが、書籍のグラフのようにうまく収束しませんでした。 試しに以下の通りOptimizerをSGD -> Adamに変更したら書籍の収束に近くなりました。 optimizer = optim.Adam(model.parameters(), lr=0.01) #optimizer = optim.SGD(model.parameters(), lr=0.01)...