Files
hao-ai-lab-FastVideo/scripts/train.sh
T
LinB203 1b99cf2789 add sample script
Former-commit-id: 9ce847651bb6dba28c51ce5807e222ae2902a78a
2024-03-06 23:23:17 +08:00

7 lines
364 B
Bash

torchrun --nproc_per_node=8 --master_port=29501 opensora/train/train.py \
--model Latte-XL/122 --dataset sky \
--ae stabilityai/sd-vae-ft-mse \
--data-path /remote-home/yeyang/sky_timelapse/sky_train/ --extras 1 \
--sample-rate 3 --num-frames 16 --max-image-size 256 \
--epochs 14000 --global-batch-size 40 --lr 1e-4 \
--ckpt-every 1000 --log-every 50