File size: 438 Bytes
bab971b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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/