2025年11月28日
pytorch中num_worker的工作原理
摘要： DataLoader 的核心作用是 “并行加载数据”，避免训练时主进程等待数据（IO 阻塞）。num_workers 定义了专门用于数据加载的子进程数量： num_workers=0：单进程模式（默认），数据加载和模型训练都在主进程中执行，无额外子进程，CPU 内存占用最低，但数据加载速度最慢（训练 阅读全文
posted @ 2025-11-28 17:33 Picassooo 阅读(10) 评论(0) 推荐(0)