TadaoYamaoka https://blog.hatena.ne.jp/TadaoYamaoka/ TadaoYamaokaの開発日記 https://tadaoyamaoka.hatenablog.com/ 拡散モデル 将棋AI 前回、拡散モデルで将棋の方策を学習できることを確認した。今回は、マルチGPUで学習できるように学習処理をPyTorch Lightningで実装し直す。 LightningCLI LightningCLIを使うと、ハイパーパラメータやオプティマイザの設定をconfigファイルに記述できて便利である。 モデルクラスと、データクラスを定義すると、以下のように記述するだけで、学習部を実装できる。 LightningCLI(DiffusionPolicy, MyDataModule) 実行するときは、サブコマンド「fit」を指定して python train_lightning.py fit --co… 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Ftadaoyamaoka.hatenablog.com%2Fentry%2F2023%2F09%2F10%2F235721" title="拡散モデルで将棋の方策を学習する その３ - TadaoYamaokaの開発日記" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> Hatena Blog https://hatena.blog 2023-09-10 23:57:21 rich https://tadaoyamaoka.hatenablog.com/entry/2023/09/10/235721 1.0 100%