From fddb3f4bf70f8efa2f4a8a55b7fc2bde56f5a30c Mon Sep 17 00:00:00 2001 From: zhengzangw Date: Tue, 23 Apr 2024 12:15:09 +0000 Subject: [PATCH] update config --- configs/opensora-v1-1/train/stage3-hq.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/configs/opensora-v1-1/train/stage3-hq.py b/configs/opensora-v1-1/train/stage3-hq.py index 470b2df..b12699b 100644 --- a/configs/opensora-v1-1/train/stage3-hq.py +++ b/configs/opensora-v1-1/train/stage3-hq.py @@ -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 = {