赞
踩
该问题的产生是由于windows下多线程的问题,和DataLoader类有关,具体细节点这里Fix memory leak when using multiple workers on Windows。
解决方案:
在utils路径下找到datasets.py这个文件,将里面的第81行里面的参数nw改完0就可以了。
self.num_workers等于0的情况,也就是不采用多进程进行数据读取 。
这样运行train.py函数训练自己的模型了就不会出错了。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。