Change default batch size to 2

configs/config.json

@@ -8,7 +8,7 @@
     "learning_rate": 0.0002,
     "betas": [0.8, 0.99],
     "eps": 1e-9,
-    "batch_size": 4,
+    "batch_size": 2,
     "bf16_run": false,
     "lr_decay": 0.99995,
     "segment_size": 16384,
@@ -67,5 +67,5 @@
     "use_spectral_norm": false,
     "gin_channels": 256
   },
-  "version": "2.1.1"
+  "version": "2.2"
 }

configs/configs_jp_extra.json

@@ -7,7 +7,7 @@
     "learning_rate": 0.0001,
     "betas": [0.8, 0.99],
     "eps": 1e-9,
-    "batch_size": 24,
+    "batch_size": 2,
     "bf16_run": false,
     "fp16_run": false,
     "lr_decay": 0.99996,
@@ -74,5 +74,5 @@
       "initial_channel": 64
     }
   },
-  "version": "2.1.1-JP-Extra"
+  "version": "2.2-JP-Extra"
 }

docs/CHANGELOG.md

@@ -4,6 +4,7 @@
 
 ### 変更・機能追加
 - bfloat16オプションはデメリットしか無さそうということで、常にオフで学習するよう変更
+- バッチサイズのデフォルトを4から2に変更。学習が異様に遅い場合はバッチサイズを下げて、VRAMに余裕があれば上げてください。
 - 学習の際の検証データ数をデフォルトで0に変更し、また検証データ数を学習用WebUIで指定できるようにした
 - Tensorboardのログ間隔を学習用WebUIで指定できるようにした
 

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,