nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-14 18:25:35 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

update config

Browse source
This commit is contained in:
zhengzangw 2024-04-23 12:15:09 +00:00
parent 1e702a3088
commit fddb3f4bf7
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
configs/opensora-v1-1/train/stage3-hq.py
View file

@ -8,14 +8,14 @@ dataset = dict(
transform_name="resize_crop",
)
bucket_config = { # 18s/it
"144p": {1: (1.0, 500), 16: (1.0, 36), 32: (1.0, 18), 64: (1.0, 9), 128: (1.0, 4)},
"256": {1: (0.8, 300), 16: (0.5, 24), 32: (0.5, 12), 64: (0.5, 7), 128: (0.8, 4)},
"240p": {1: (0.8, 300), 16: (0.5, 24), 32: (0.5, 12), 64: (0.5, 7), 128: (0.5, 4)},
"360p": {1: (0.5, 150), 16: (0.5, 12), 32: (0.5, 6), 64: (0.5, 3), 128: (0.5, 1)},
"512": {1: (0.5, 150), 16: (0.5, 12), 32: (0.5, 6), 64: (0.5, 3), 128: (0.8, 1)},
"480p": {1: (0.4, 100), 16: (0.6, 8), 32: (0.6, 4), 64: (0.6, 2), 128: (0.8, 1)},
"720p": {1: (0.4, 50), 16: (0.6, 4), 32: (0.6, 2), 64: (0.6, 1), 96: (0.0, None)},
"1024": {1: (0.3, 50)},
"144p": {1: (1.0, 200), 16: (1.0, 36), 32: (1.0, 18), 64: (1.0, 9), 128: (1.0, 4)},
"256": {1: (0.8, 200), 16: (0.5, 24), 32: (0.5, 12), 64: (0.5, 6), 128: (0.8, 4)},
"240p": {1: (0.8, 200), 16: (0.5, 24), 32: (0.5, 12), 64: (0.5, 6), 128: (0.5, 4)},
"360p": {1: (0.5, 120), 16: (0.5, 11), 32: (0.5, 5), 64: (0.5, 2), 128: (0.5, 1)},
"512": {1: (0.5, 120), 16: (0.5, 11), 32: (0.5, 5), 64: (0.5, 2), 128: (0.8, 1)},
"480p": {1: (0.4, 80), 16: (0.6, 8), 32: (0.6, 4), 64: (0.6, 2), 128: (0.0, None)},
"720p": {1: (0.4, 40), 16: (0.6, 3), 32: (0.6, 1), 96: (0.0, None)},
"1024": {1: (0.3, 40)},
"1080p": {1: (0.3, 20)},
}
mask_ratios = {