runTimeError与多线程

在学习pytorch官网上tutorial部分时,STN网络的代码是很重要的一部分内容。然而,将代码在自己的电脑中运行时,却出现了runTimeError的报错,具体第二幅图所示。
原因是dataset内部可能会开启多线程。事实上在这个代码中dataset的确开启了多线程,体现在语句“
numworker = 4”中。将4改为0即可。(改为1应该也是可以的,不过没试过,0和1理论上在这个情况里是等效的)。
最后一幅图是运行成功的效果图。
runTimeError与多线程_第1张图片runTimeError与多线程_第2张图片![在这里插入图片描述](https://img-blog.csdnimg.cn/20190405192659630.jpg?x-oss-process=image/watermark,type_ZmrunTimeError与多线程_第3张图片FuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L

你可能感兴趣的:(runTimeError与多线程)