原因: cpu和gpu 通讯太慢了。

解决办法:

在dataloader 里设置 num_workers=16(或你想要的参数)

 参考:https://www.jb51.net/article/213809.htm

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐