吴吴大庸
updated the project based on https://huggingface.co/spaces/LanguageBind/Open-Sora-Plan-v1.1.0/tree/main
a5130bc
| python opensora/train/train_causalvae.py \ | |
| --exp_name "exp_name" \ | |
| --batch_size 1 \ | |
| --precision bf16 \ | |
| --max_steps 40000 \ | |
| --save_steps 100 \ | |
| --output_dir results/causalvae_ \ | |
| --video_path /remote-home1/dataset/data_split_tt \ | |
| --video_num_frames 17 \ | |
| --resolution 256 \ | |
| --sample_rate 1 \ | |
| --n_nodes 1 \ | |
| --devices 1 \ | |
| --num_workers 8 \ | |
| --load_from_checkpoint ./results/pretrained_488/ |