tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Improve: vram usage guide

Browse Source
This commit is contained in:
litagin02
2024-02-09 17:42:47 +09:00
parent 0a49d1ee00
commit 862a719e2f
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/CHANGELOG.md
View File

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

2
webui_train.py
View File

@@ -435,7 +435,7 @@ if __name__ == "__main__":
) )
batch_size = gr.Slider( batch_size = gr.Slider(
label="バッチサイズ", label="バッチサイズ",
info="速度が異様に遅い場合は小さく、VRAMに余裕があれば大きくしてください", info="速度が異様に遅い場合は小さく、VRAMに余裕があれば大きくしてください。使用量目安: 1: 6GB, 2: 8GB, 3: 10GB, 4: 12GB",
value=2, value=2,
minimum=1, minimum=1,
maximum=64, maximum=64,