Pytorch dataloader getitem 异常报错处理

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

def __getitem__(self, item):
    

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

你可能感兴趣的:(pytorch,人工智能,python)