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