当前位置:
首页 >
[W pthreadpool-cpp.cc:90] Warning:Leaking Caffe2 thread-pool after fork.(function pthreadpool)
发布时间:2024/3/7
414
豆豆
生活随笔
收集整理的这篇文章主要介绍了
[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**)
改为
总结
以上是生活随笔为你收集整理的[W pthreadpool-cpp.cc:90] Warning:Leaking Caffe2 thread-pool after fork.(function pthreadpool)的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇: Chapter5 初始化(Initial
- 下一篇: 初识Caffeine