2021年7月30日
(十)pytorch多线程训练，DataLoader的num_works参数设置
摘要： 一、概述 数据集较小时（小于2W）建议num_works不用管默认就行，因为用了反而比没用慢。当数据集较大时建议采用，num_works一般设置为（CPU线程数+-1）为最佳，可以用以下代码找出最佳num_works（注意windows用户如果要使用多核多线程必须把训练放在if __name__ = 阅读全文
posted @ 2021-07-30 13:31 jasonzhangxianrong 阅读(4421) 评论(0) 推荐(0) 编辑