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

Merge branch 'dev/v1.1' of github.com:hpcaitech/Open-Sora-dev into dev/v1.1

Browse source
This commit is contained in:
zhengzangw 2024-04-21 12:47:24 +00:00
commit 43485edb69
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
configs/opensora-v1-1/train/stage1-2.py
View file

@ -9,11 +9,11 @@ dataset = dict(
)
# IMG: 1024 (20%) 512 (30%) 256 (50%) drop (50%)
bucket_config = { # 1s/it
"144p": {16: (1.0, 8), 32: (1.0, 4), 64: (1.0, 2), 128: (1.0, 1)},
"256": {1: (0.5, 32), 16: (0.5, 4), 32: (0.5, 2), 64: (0.5, 1), 128: (0.0, None)},
"240p": {16: (0.3, 3), 32: (0.3, 1), 64: (0.0, None)},
"512": {1: (0.4, 16)},
"1024": {1: (0.3, 4)},
"144p": {1: (0.5, 48), 16: (1.0, 6), 32: (1.0, 3), 96: (1.0, 1)},
"256": {1: (0.5, 24), 16: (0.5, 3), 48: (0.5, 1), 64: (0.0, None)},
"240p": {16: (0.3, 2), 32: (0.3, 1), 64: (0.0, None)},
"512": {1: (0.4, 12)},
"1024": {1: (0.3, 3)},
}
mask_ratios = {
"mask_no": 0.75,