nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 05:13:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[config] update training config

Browse source
This commit is contained in:
zhengzangw 2024-06-05 07:42:45 +00:00
parent c04f157fec
commit d028417a61
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
configs/opensora-v1-2/train/stage2.py
View file

@ -14,7 +14,7 @@ bucket_config = { # 12s/it
"360p": {1: (0.5, 141), 51: (0.15, 8), 102: ((0.3, 0.5), 4), 204: ((0.3, 1.0), 2), 408: ((0.5, 0.5), 1)},
"512": {1: (0.4, 141), 51: (0.15, 8), 102: ((0.2, 0.4), 4), 204: ((0.2, 1.0), 2), 408: ((0.4, 0.5), 1)},
# ---
"480p": {1: (0.5, 89), 51: (0.1, 5), 102: (0.1, 2), 204: (0.05, 1)},
"480p": {1: (0.5, 89), 51: (0.2, 5), 102: (0.2, 2), 204: (0.1, 1)},
# ---
"720p": {1: (0.1, 36), 51: (0.03, 1)},
"1024": {1: (0.1, 36), 51: (0.02, 1)},