迭代训练集,随机抽取batch_size数量的图片,报错索引超出范围index out of range
在训练SRGAN模型时,训练到一半突然终止并报错索引越界,排查发现是在迭代dataloder时,构造数据集的函数中有这样一句:这句的作用是按照排列好的顺序读取图片,打印index语句的时候,打印了一部分,之后出现了索引越界错误,应该是其中某些文件的索引有空,或者超出范围,解决方案是跳过这些索引的文件,所以使用try,except,把这两句框起来,没错的时候正常运行,有错的时候跳过当前索引的文件,就