2020年6月11日
pytorch（六）一些训练技巧
摘要： 1.dataloader训练时的num_worker设置（推荐设置为1） num_workers = 0 只用主进程main process （训练程序的进程)来加载数据。主进程完成一个batch的前向后向传播，再去disk搬运下一个batch到cpu，然后再转移到GPU。 num_workers 阅读全文
posted @ 2020-06-11 14:48 Parallax 阅读(343) 评论(0) 推荐(0) 编辑