runTimeError與多線程

在學習pytorch官網上tutorial部分時，STN網絡的代碼是很重要的一部分內容。然而，將代碼在自己的電腦中運行時，卻出現了runTimeError的報錯，具體第二幅圖所示。 原因是dataset內部可能會開啓多線程。事實上在這個代碼中dataset的確開啓了多線程，體現在語句「 numworker = 4」中。將4改爲0即可。（改爲1應該也是可以的，不過沒試過，0和1理論上在這個情況裏是等

>>阅读原文<<