Improve: vram usage guide

2024-02-09 17:42:47 +09:00
parent 0a49d1ee00
commit 862a719e2f
2 changed files with 2 additions and 2 deletions
--- 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で指定できるようにした

--- 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,