From 862a719e2f65953b3dcd29ce86d3532b2e27ba47 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Fri, 9 Feb 2024 17:42:47 +0900 Subject: [PATCH] Improve: vram usage guide --- docs/CHANGELOG.md | 2 +- webui_train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md index 38fd542..79a29a5 100644 --- a/docs/CHANGELOG.md +++ b/docs/CHANGELOG.md @@ -4,7 +4,7 @@ ### 変更・機能追加 - bfloat16オプションはデメリットしか無さそうということで、常にオフで学習するよう変更 -- バッチサイズのデフォルトを4から2に変更。学習が異様に遅い場合はバッチサイズを下げて、VRAMに余裕があれば上げてください。 +- バッチサイズのデフォルトを4から2に変更。学習が遅い場合はバッチサイズを下げて試してみて、VRAMに余裕があれば上げてください。バッチサイズごとのVRAM使用量目安は、1: 6GB, 2: 8GB, 3: 10GB, 4: 12GB くらいのようです。 - 学習の際の検証データ数をデフォルトで0に変更し、また検証データ数を学習用WebUIで指定できるようにした - Tensorboardのログ間隔を学習用WebUIで指定できるようにした diff --git a/webui_train.py b/webui_train.py index 3ea4bb4..ddae8f4 100644 --- a/webui_train.py +++ b/webui_train.py @@ -435,7 +435,7 @@ if __name__ == "__main__": ) batch_size = gr.Slider( label="バッチサイズ", - info="速度が異様に遅い場合は小さく、VRAMに余裕があれば大きくしてください", + info="速度が異様に遅い場合は小さく、VRAMに余裕があれば大きくしてください。使用量目安: 1: 6GB, 2: 8GB, 3: 10GB, 4: 12GB", value=2, minimum=1, maximum=64,