Change default batch size to 2

2024-02-09 16:01:30 +09:00
parent 5876624f01
commit 527908626b
4 changed files with 8 additions and 6 deletions
--- a/webui_train.py
+++ b/webui_train.py
@@ -435,7 +435,8 @@ if __name__ == "__main__":
                )
                batch_size = gr.Slider(
                    label="バッチサイズ",
-                    value=4,
+                    info="速度が異様に遅い場合は小さく、VRAMに余裕があれば大きくしてください",
+                    value=2,
                    minimum=1,
                    maximum=64,
                    step=1,
@@ -519,7 +520,7 @@ if __name__ == "__main__":
                    )
                    batch_size_manual = gr.Slider(
                        label="バッチサイズ",
-                        value=4,
+                        value=2,
                        minimum=1,
                        maximum=64,
                        step=1,