Pytorch dataloader getitem 异常报错处理

当在修改别的sota模型代码的时候，我们的数据集格式经常需要转换，这个时候容易出现问题，怎样在getitem中写一个异常报错使得程序不断掉，不然让它跑一晚上结果只跑了几个小时，真的心累
def __getitem__(self, item):
    

    try:
    
    except:
        #判断一下边界，也可能-1
        return self.__getitem__(item+1)

原文链接：https://blog.csdn.net/hxxjxw/article/details/123148108

你可能也喜欢