在实践中 数据读取经常是训练的性能瓶颈 特别当模型简单或者计算硬件性能较高时。Pytorch的DataLoader中一个很方便的功能是允许使用多进程来加速数据读取。具体使用时设定其中的num_workers参数来指定读取数据的进程数。
上一篇 「深度学习一遍过」必修27:基于Mask
下一篇 Python 语句
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号