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:
parent
dc1ff288e8
commit
3be569a31e
|
|
@ -24,7 +24,7 @@ import csv
|
||||||
import torch
|
import torch
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
|
|
||||||
N_SAMPLES = 4 # 生成 4 条假样本
|
N_SAMPLES = 16 # 8 GPU × 2 batch/GPU,保证每卡至少有数据
|
||||||
BASE_DIR = "/data/train-input/smoke_test"
|
BASE_DIR = "/data/train-input/smoke_test"
|
||||||
NULL_DIR = "/mnt/ddn/sora/tmp_load"
|
NULL_DIR = "/mnt/ddn/sora/tmp_load"
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue