pytorch鎖死在dataloader（訓練時卡死）

1.問題描述 2.解決方案 （1）Dataloader裏面不用cv2.imread進行讀取圖片，用cv2.imread還會帶來一系列的不方便，比如不能結合torchvision進行數據增強，所以最好用PIL 裏面的Image.open來讀圖片。（並不適用本例） （2）將DataLoader 裏面的參變量num_workers設置爲0，但會導致數據的讀取很慢，拖慢整個模型的訓練。（並不適用本例） （

>>阅读原文<<