当在修改别的sota模型代码的时候,我们的数据集格式经常需要转换,这个时候容易出现问题,怎样在getitem中写一个异常报错使得程序不断掉,不然让它跑一晚上结果只跑了几个小时,真的心累
def __getitem__(self, item): try: except: #判断一下边界,也可能-1 return self.__getitem__(item+1)
Pytorch dataloader getitem 异常报错处理
于 2022-02-26 12:26:24 首次发布
当在修改别的sota模型代码的时候,我们的数据集格式经常需要转换,这个时候容易出现问题,怎样在getitem中写一个异常报错使得程序不断掉,不然让它跑一晚上结果只跑了几个小时,真的心累
def __getitem__(self, item): try: except: #判断一下边界,也可能-1 return self.__getitem__(item+1)