原因分析:报错的原因让人很意想不到,在定义Dataset的时候,是读取.txt文件中的文件名进行数据划分的,在.txt文件中不小心多打了两行空行,这里也会将空行读进去,变成None,导致DataLoader在取数据的时候出问题。
解决方法:将空行删去即可。
原因分析:报错的原因让人很意想不到,在定义Dataset的时候,是读取.txt文件中的文件名进行数据划分的,在.txt文件中不小心多打了两行空行,这里也会将空行读进去,变成None,导致DataLoader在取数据的时候出问题。
解决方法:将空行删去即可。