pytorch 出现Pytorch TypeError: batch must contain tensors, numbers, dicts or lists错误,应该是dataloader的错误。
可以加上如下几句:
if self.is_transfrom:
img_data=self.is_transfrom(img_data)
return img_data,lab
参考博客 https://blog.csdn.net/qq_36285997/article/details/82463959