runTimeError与多线程

在学习pytorch官网上tutorial部分时,STN网络的代码是很重要的一部分内容。然而,将代码在自己的电脑中运行时,却出现了runTimeError的报错,具体第二幅图所示。 原因是dataset内部可能会开启多线程。事实上在这个代码中dataset的确开启了多线程,体现在语句“ numworker = 4”中。将4改为0即可。(改为1应该也是可以的,不过没试过,0和1理论上在这个情况里是等
相关文章
相关标签/搜索