qqingzheng
71541c766a
Former-commit-id: 1d9358e84d12863681b396c7621ae7244caa4dc2 [formerly b801491889b3abad1211225d7f07af3ae0389895] Former-commit-id: 9e6a59ae2f246b50ace0a033cbd6548660241204
15 lines
437 B
Bash
15 lines
437 B
Bash
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 9 \
|
|
--resolution 432 \
|
|
--sample_rate 1 \
|
|
--n_nodes 1 \
|
|
--devices 1 \
|
|
--num_workers 8 \
|
|
--load_from_checkpoint ./results/pretrained_488/ |