qqingzheng
a4fe6a634d
Former-commit-id: eeb00ac046692bc97a064ec1b47c994436798ab1 [formerly 6d0fd85e1b13a7824b8181ba9ccebec2a3c76d40] Former-commit-id: 1e8e669ad9967aeeeb7622bc04a0759f348cb26f
15 lines
438 B
Bash
15 lines
438 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 17 \
|
|
--resolution 256 \
|
|
--sample_rate 1 \
|
|
--n_nodes 1 \
|
|
--devices 1 \
|
|
--num_workers 8 \
|
|
--load_from_checkpoint ./results/pretrained_488/ |