diff --git a/configs/opensora-v1-2/train/stage2.py b/configs/opensora-v1-2/train/stage2.py index 94e6975..8620066 100644 --- a/configs/opensora-v1-2/train/stage2.py +++ b/configs/opensora-v1-2/train/stage2.py @@ -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"