今天在另一台电脑中首次跑LPRnet出现了一个这样的错误, torch.utils.data 下没有collate这个包。
因为是在虚拟环境中运行,所以记录一下在虚拟环境中怎么解决这样的问题。
虚拟环境的torch.utils.data路径到底在哪?如图
“learn”是自己的环境名(没有和第一张图对应上,当时截图没注意,如果和第一张图对上那么就应该是“cch”),python3.8是创建虚拟环境时的python版本,如果创建的是3.7版本那对应的目录就是python3.7。
将_utils目录下的collate.py复制到上一级data目录中,问题解决。
解决linux虚拟环境上的AttributeError: module ‘torch.utils.data’ has no attribute ‘collate’问题
最新推荐文章于 2023-05-08 23:44:06 发布
在运行LPRnet时遇到torch.utils.data缺失collate包的错误。解决方案是在虚拟环境的torch.utils.data目录下,将_utils目录中的collate.py文件复制到data目录中,以此解决依赖问题。此方法适用于不同Python版本的虚拟环境。
摘要由CSDN通过智能技术生成