From 68778cda9f36c6990b579bae13bfc1ff30fd0374 Mon Sep 17 00:00:00 2001 From: zhengzangw Date: Fri, 7 Jun 2024 05:49:34 +0000 Subject: [PATCH 1/2] update train config --- scripts/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/scripts/train.py b/scripts/train.py index d10e2fb..777d20c 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -1,4 +1,3 @@ -import gc import os from copy import deepcopy from datetime import timedelta @@ -69,7 +68,7 @@ def main(): if coordinator.is_master(): tb_writer = create_tensorboard_writer(exp_dir) if cfg.get("wandb", False): - wandb.init(project="minisora", name=exp_name, config=cfg.to_dict(), dir="./outputs/wandb") + wandb.init(project="Open-Sora", name=exp_name, config=cfg.to_dict(), dir="./outputs/wandb") # == init ColossalAI booster == plugin = create_colossalai_plugin( From a521735c5f3caa1c5947e3fb331a4cb11e92818b Mon Sep 17 00:00:00 2001 From: pxy Date: Fri, 7 Jun 2024 07:25:34 +0000 Subject: [PATCH 2/2] update cut --- tools/scene_cut/cut.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/scene_cut/cut.py b/tools/scene_cut/cut.py index 2e7b32c..ed958c9 100644 --- a/tools/scene_cut/cut.py +++ b/tools/scene_cut/cut.py @@ -146,7 +146,7 @@ def parse_args(): parser.add_argument("--max_seconds", type=float, default=None, help='if not None, clip longer than max_seconds is truncated') parser.add_argument("--target_fps", type=int, default=None, help='target fps of clips') - parser.add_argument("--shorter_size", type=int, default=None, help='resize the shorter size by keeping ratio; will not do upscale') + parser.add_argument("--shorter_size", type=int, default=1080, help='resize the shorter size by keeping ratio; will not do upscale') parser.add_argument("--num_workers", type=int, default=None, help='#workers for pandarallel') args = parser.parse_args()