PyTorchで画像分類(その4)

前回は学習に必要な損失関数と最適化アルゴリズムを作成したので、今回は実際に学習をします。 題材は前回までと同じkaggleの犬/猫の画像分類コンペを使います。 学習 早速ですが、実装します。 ソースコードのコメントに簡単な説明を入れました。 Kerasの場合、損失関数と最適化アルゴリズムをcompileメソッドに渡してfitメソッドを呼べば学習できてしまいま […]