Pytorch训练错误:AttributeError: Can't pickle local object 'main..'

在基于Pytorch进行深度学习模型训练时,有可能出现如下错误:

AttributeError: Can't pickle local object 'main.<locals>.<lambda>'

当出现该错误时,只需要将训练配置的num_workers参数修改为0即可

阅读剩余
THE END