Update train_256.sh

Former-commit-id: a2dfc9a8c70845446e731c88acd222e9b2db9122
This commit is contained in:
lb203
2024-03-12 14:21:14 +08:00
committed by GitHub
parent a0f4a001a3
commit c8811aa2d5
+1 -1
View File
@@ -8,7 +8,7 @@ accelerate launch \
--data-path /remote-home/yeyang/sky_timelapse/sky_train/ \
--extras 1 \
--sample-rate 3 \
--num-frames 16 \
--num-frames 128 \
--max-image-size 256 \
--max-train-steps 1000000 \
--local-batch-size 5 \