分类：锁死dataloader

2021
07-21

pytorch锁死在dataloader(训练时卡死)

1.问题描述2.解决方案（1）Dataloader里面不用cv2.imread进行读取图片，用cv2.imread还会带来一系列的不方便，比如不能结合torchvision进行数据增强，所以最好用PIL里面的Image.open来读图片。（并不适用本例）（2）将DataLoader里面的参变量num_workers设置为0，但会导致数据的读取很慢，拖慢整个模型的训练。（并不适用本例）（3）如果用了cv2.imread，不想改代码的，那就加两条语句，来关闭Opencv的多线程：cv2.setNumThreads... 继续阅读 >