add sample script
Former-commit-id: 9ce847651bb6dba28c51ce5807e222ae2902a78a
This commit is contained in:
@@ -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
@@ -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
|
||||
Reference in New Issue
Block a user