@@ -322,7 +322,7 @@ def main():
# 双GPU处理参数
batch_size = 4 # 批处理大小
- workers_per_gpu = 2 # 每个GPU的worker数量（24GB GPU推荐2个）
+ workers_per_gpu = 1 # 每个GPU的worker数量（24GB GPU推荐2个）
# 确保输出目录存在
print(f"输出目录: {Path(output_dir).absolute()}")
