| nohup torchrun \ | |
| --nnodes=1 \ | |
| --nproc_per_node=4 \ | |
| --rdzv_endpoint=localhost:29739 \ | |
| train_rectified_noise.py \ | |
| --depth 2 \ | |
| --results-dir results_256_gvp_disp \ | |
| --data-path /gemini/platform/public/zhaozy/hzh/datasets/Imagenet/train/ \ | |
| --ckpt /gemini/space/zhaozy/zhy/gzy_new/Noise_Matching/SiT_clean_256_GVP/base.pt \ | |
| --num-classes 1000 \ | |
| --path-type GVP \ | |
| --prediction velocity \ | |
| --disp \ | |
| > w_training1.log 2>&1 & | |