nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-10 21:01:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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

Browse source
This commit is contained in:
zhengzangw 2024-06-22 13:30:28 +00:00
commit 6893e9464b
1 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -60,19 +60,21 @@ scheduler = dict(
)
# Mask settings
# 25%
mask_ratios = {
"random": 0.05,
"intepolate": 0.005,
"quarter_random": 0.005,
"quarter_head": 0.005,
"quarter_tail": 0.005,
"quarter_head_tail": 0.005,
"image_random": 0.025,
"image_head": 0.05,
"image_tail": 0.025,
"image_head_tail": 0.025,
"random": 0.005,
"intepolate": 0.002,
"quarter_random": 0.007,
"quarter_head": 0.002,
"quarter_tail": 0.002,
"quarter_head_tail": 0.002,
"image_random": 0.0,
"image_head": 0.22,
"image_tail": 0.005,
"image_head_tail": 0.005,
}
# Log settings
seed = 42
outputs = "outputs"