欢迎访问 生活随笔!

凯发k8官方网

当前位置: 凯发k8官方网 > 人工智能 > caffe >内容正文

caffe

解决warning: leaking caffe2 thread-凯发k8官方网

发布时间:2024/3/7 caffe 71 豆豆
凯发k8官方网 收集整理的这篇文章主要介绍了 解决warning: leaking caffe2 thread-pool after fork 小编觉得挺不错的,现在分享给大家,帮大家做个参考.

问题:使用pytorch中dataloader时出现警告:

epoch 2: [w pthreadpool-cpp.cc:88] warning: leaking caffe2 thread-pool after fork. (function pthreadpool) [w pthreadpool-cpp.cc:88] warning: leaking caffe2 thread-pool after fork. (function pthreadpool)

解决:设置pin_memory=false

总结

以上是凯发k8官方网为你收集整理的解决warning: leaking caffe2 thread-pool after fork的全部内容,希望文章能够帮你解决所遇到的问题。

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

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