Visual Studio Code远程调试pytorch模型训练时,报错如下:
解决办法,在.py文件头添加如下语句:
更新:
如果你的工程是基于pytorch的,那么检查自己的dataLoader,是否使用了num_workers参数。当使用该参数时,可能会报上述错误。解决办法是将num_workers设置为0
Visual Studio Code远程调试pytorch模型训练时,报错如下:
解决办法,在.py文件头添加如下语句:
更新:
如果你的工程是基于pytorch的,那么检查自己的dataLoader,是否使用了num_workers参数。当使用该参数时,可能会报上述错误。解决办法是将num_workers设置为0