add sample script

Former-commit-id: 9ce847651bb6dba28c51ce5807e222ae2902a78a
This commit is contained in:
LinB203
2024-03-06 23:23:17 +08:00
parent ae2646e213
commit 1b99cf2789
11 changed files with 391 additions and 146 deletions
+4
View File
@@ -0,0 +1,4 @@
python opensora/sample/sample.py \
--model Latte-XL/122 --ae stabilityai/sd-vae-ft-mse \
--ckpt results/047-Latte-XL-122/checkpoints/0005000.pt --extras 1 \
--fps 10 --num-frames 16 --image-size 256
+6 -6
View File
@@ -1,7 +1,7 @@
torchrun --nproc_per_node=1 --master_port=29501 opensora/train/train.py \
--model DiT-XL/122 --dataset ucf101\
--ae ucf101_stride4x4x4 \
--data-path /remote-home/yeyang/UCF-101 --num-classes 101 \
--sample-rate 2 --num-frames 8 --max-image-size 128 --clip-grad-norm 1 \
--epochs 14000 --global-batch-size 2 --lr 1e-4 \
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