欢迎访问 生活随笔!

ag凯发k8国际

当前位置: ag凯发k8国际 > 人工智能 > caffe >内容正文

caffe

[w pthreadpool-ag凯发k8国际

发布时间:2024/3/7 caffe 96 豆豆
ag凯发k8国际 收集整理的这篇文章主要介绍了 [w pthreadpool-cpp.cc:90] warning:leaking caffe2 thread-pool after fork.(function pthreadpool) 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

训练时在每一个epoch结束后就会出现好几行这个警告:

解释一:

原因是:线程撕裂,出现了警告,警告数量为设置的线程数量,32个:

大家说无关紧要,说是如果把线程数改小一些,就不会有警告了,但是会影响运行速度,所以可以当作不存在。pytorch版本1.10之后就会修复了,问题不大。

解释二:

找到代码中的:

torch.utils.data.dataloader(val_dataset, batch_size=args.batch_size, shuffle=true, num_workers=args.num_workers, pin_memory=**true**)


改为

torch.utils.data.dataloader(val_dataset, batch_size=args.batch_size, shuffle=true, num_workers=args.num_workers, pin_memory=**false**)

 

总结

以上是ag凯发k8国际为你收集整理的[w pthreadpool-cpp.cc:90] warning:leaking caffe2 thread-pool after fork.(function pthreadpool)的全部内容,希望文章能够帮你解决所遇到的问题。

如果觉得ag凯发k8国际网站内容还不错,欢迎将ag凯发k8国际推荐给好友。

  • 上一篇:
  • 下一篇:
网站地图