nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 13:14:44 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Merge branch 'dev/v1.2' into feature/parallel_eval

Browse source
This commit is contained in:
Shen-Chenhui 2024-06-08 03:12:55 +00:00
commit 2e180ddf25
2 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
scripts/train.py
View file

@ -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(

2
tools/scene_cut/cut.py
View file

@ -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()