From 3be569a31e476c9e4f25d6776e9cb6d05459b3fb Mon Sep 17 00:00:00 2001
From: hailin <hailin.zhao@gdzx.xyz>
Date: Fri, 6 Mar 2026 03:23:26 -0800
Subject: [PATCH] fix: increase smoke test samples to 16 for 8-GPU distribution

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 scripts/diffusion/gen_smoke_data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/diffusion/gen_smoke_data.py b/scripts/diffusion/gen_smoke_data.py
index 6a9634e..679ce99 100644
--- a/scripts/diffusion/gen_smoke_data.py
+++ b/scripts/diffusion/gen_smoke_data.py
@@ -24,7 +24,7 @@ import csv
 import torch
 from PIL import Image
 
-N_SAMPLES = 4           # 生成 4 条假样本
+N_SAMPLES = 16          # 8 GPU × 2 batch/GPU，保证每卡至少有数据
 BASE_DIR  = "/data/train-input/smoke_test"
 NULL_DIR  = "/mnt/ddn/sora/tmp_load"