首页 > 其他 > 详细

关于Dataloader中的num_workers

时间:2019-11-15 20:28:58      阅读:710      评论:0      收藏:0      [点我收藏+]

num_workers通过影响数据加载速度,从而影响训练速度

在调整num_workers的过程中,发现训练速度并没有变化

原因在于:

num_workers是加载数据(batch)的线程数目

当加载batch的时间 < 数据训练的时间

  GPU每次训练完都可以直接从CPU中取到next batch的数据

  无需额外的等待,因此也不需要多余的worker,即使增加worker也不会影响训练速度

当加载batch的时间 > 数据训练的时间

  GPU每次训练完都需要等待CPU完成数据的载入

  若增加worker,即使worker_1还未就绪,GPU也可以取worker_2的数据来训练

仅限单线程训练情况

 

Reference:

https://www.cnblogs.com/hesse-summer/p/11343870.html

https://blog.csdn.net/attackkk/article/details/101914509

关于Dataloader中的num_workers

原文:https://www.cnblogs.com/JunzhaoLiang/p/11869268.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!