在实践中,数据读取经常是训练的性能瓶颈,特别当模型简单或者计算硬件性能较高时。Pytorch的DataLoader中一个很方便的功能是允许使用多进程来加速数据读取。具体使用时设定其中的num_workers参数来指定读取数据的进程数。
Pytorch中DataLoader相关操作
最新推荐文章于 2024-03-19 21:02:38 发布
在实践中,数据读取经常是训练的性能瓶颈,特别当模型简单或者计算硬件性能较高时。Pytorch的DataLoader中一个很方便的功能是允许使用多进程来加速数据读取。具体使用时设定其中的num_workers参数来指定读取数据的进程数。