2022年1月3日
pytorch num_worker>0 出错或者卡住不动了原因及解决方案
摘要： 我的代码在data_loader中使用了opencv 来读取图片。 但是据说opencv3.4多线程是有bug的，会跟pytorch的多线程冲突。 num_worker>0 则代码崩溃，改成num_worker=0 则很慢. ##解决方案： import cv2 cv2.setNumThreads( 阅读全文
posted @ 2022-01-03 23:51 PengPanda 阅读(929) 评论(0) 推荐(0)