File size: 354 Bytes
8b0b874 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | torchrun \
--nnodes=1 \
--nproc_per_node=8 \
train.py \
--model SiT-S/2 \
--epochs=800 \
--data-path /home/nvidia/datasets/imagenet-1k/train \
--wandb \
--global-batch-size=1024 \
--run-name 005-SiT-S-2-Linear-velocity-None \
--ckpt /home/nvidia/SiT-Complementary/results/005-SiT-S-2-Linear-velocity-None/checkpoints/0550000.pt
# batch_size x 4, lr x 2
|