hailin
/
mysora
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: increase smoke test samples to 16 for 8-GPU distribution 

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
hailin 2026-03-06 03:23:26 -08:00
parent dc1ff288e8
commit 3be569a31e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/diffusion/gen_smoke_data.py
View File

@ -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"