venoda https://blog.hatena.ne.jp/venoda/ 機械学習ともろもろ https://venoda.hatenablog.com/ PyTorch Python 1. 概要 2. モデル化の流れ 3. 準備 3.1 データ準備 3.2 前処理 3.3 Datasetの作成 3.4 DataLoaderの作成 4. Lighgningモジュールの定義 4.1. ネットワークの定義 4.2 損失関数の定義 4.3. 最適化手法の定義 5. 学習・評価 5.1. 学習処理の設定 5.2 学習と予測の実行 6. 全体のコード 1. 概要 PyTorchはモデル構築の際に柔軟にコードを書くことができますが、自由度が高いがゆえに学習用のループ処理などが複雑になりがちです。 PyTorch Lightningを使えば、PyTorchで書いていた学習用のループ処理など… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Fvenoda.hatenablog.com%2Fentry%2F2021%2F06%2F06%2F005620" title="PyTorch Lightningによる画像分類の実装 - 機械学習ともろもろ" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> https://cdn.blog.st-hatena.com/images/theme/og-image-1500.png Hatena Blog https://hatena.blog 2021-06-06 00:56:20 rich https://venoda.hatenablog.com/entry/2021/06/06/005620 1.0 100%