back/samples_ddp.sh

#!/usr/bin/env bash

# 4 卡 DDP 单路径采样（不做 dual-compare，不保存 at_tc 中间图）
CUDA_VISIBLE_DEVICES=0,1,2,3 nohup nohup torchrun \
  --nnodes=1 \
  --nproc_per_node=4 \
  --rdzv_endpoint=localhost:29110 \
  sample_from_checkpoint_ddp.py \
  --ckpt /gemini/space/zhaozy/guzhenyu/UAVFlow/UAV_Flow_base/exps/jsflow-experiment/samples/REG/exps/jsflow-experiment-0.75/checkpoints/0600000.pt \
  --out-dir ./my_samples_600k_new \
  --num-images 40000 \
  --batch-size 64 \
  --seed 0 \
  --t-c 0.75 \
  --steps-before-tc 100 \
  --steps-after-tc 50 \
  --sampler em_image_noise_before_tc \
  --cfg-scale 1.0 \
  > samples_0.75_new.log 2>&1 & 

# nohup python sample_from_checkpoint_ddp.py \
#   --ckpt /gemini/space/zhaozy/guzhenyu/UAVFlow/UAV_Flow_base/exps/jsflow-experiment/samples/REG/exps/jsflow-experiment-0.5/checkpoints/0250000.pt \
#   --out-dir ./my_samples_5 \
#   --num-images 20000 \
#   --batch-size 16 \
#   --seed 0 \
#   --t-c 0.5 \
#   --steps-before-tc 100 \
#   --steps-after-tc 50 \
#   --sampler em_image_noise_before_tc \
#   --cfg-scale 1.0 \
#   > samples_0.5.log 2>&1 &