diff --git a/configs/config.json b/configs/config.json index 128a8d9..6aa647c 100644 --- a/configs/config.json +++ b/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" } diff --git a/configs/configs_jp_extra.json b/configs/configs_jp_extra.json index 3d8ba3a..168ced6 100644 --- a/configs/configs_jp_extra.json +++ b/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" } diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md index 2d4f619..38fd542 100644 --- a/docs/CHANGELOG.md +++ b/docs/CHANGELOG.md @@ -4,6 +4,7 @@ ### 変更・機能追加 - bfloat16オプションはデメリットしか無さそうということで、常にオフで学習するよう変更 +- バッチサイズのデフォルトを4から2に変更。学習が異様に遅い場合はバッチサイズを下げて、VRAMに余裕があれば上げてください。 - 学習の際の検証データ数をデフォルトで0に変更し、また検証データ数を学習用WebUIで指定できるようにした - Tensorboardのログ間隔を学習用WebUIで指定できるようにした diff --git a/webui_train.py b/webui_train.py index 1033cd6..f401580 100644 --- 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,