Pytorch报错

报错:
TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found
原因:
没有将数据转为 torch的Tensor数据类型。在创建数据集dataset的时候把数据类型转换为tensor。

此处是dataset类中的__getitem__函数没有写返回值

通常当我们定义好Dataset并实例化dataset之后,会实例化一个DataLoader并将dataset传入其中,DataLoader的作用是拼接多个__getitem__获得的数据,返回一个batch的数据,在实例化DataLoader的时候有一个参数是collate_fn,它用来定义数据batch拼接方式
Pytorch报错_第1张图片

你可能感兴趣的:(Pytorch,python)