Pytorch——报错解决：TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, found

报错信息

TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'NoneType'>

主要原因是由于获取到batch中输入有None，反推代码，说明你的__getitem__()函数返回值为None，因此出现如下报错。

解决方法

因此，解决方法就是不要让返回值有None即可。

示例

原代码：

def prepare_train_data(self, index):"""Training data preparation.Args:index (int): Index for accessing the target data.Returns:dict: Training data dict of the corresponding index."""input_dict = self.get_data_info(index)if input_dict is None:return None     # 这里就是报错根源self.pre_pipeline(input_dict)example = self.pipeline(input_dict)return example

def prepare_train_data(self, index):"""Training data preparation.Args:index (int): Index for accessing the target data.Returns:dict: Training data dict of the corresponding index."""input_dict = self.get_data_info(index)if input_dict is None:# 如果当前为None，则取下一帧，直到不为None时return即可return self.get_data_info(index+1)self.pre_pipeline(input_dict)example = self.pipeline(input_dict)return example

参考文章：

报错记录：TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists;

Pytorch——报错解决：TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, found相关推荐

PyTorch 报错：TypeError: exceptions must derive from BaseException
项目场景: PyTorch 报错:TypeError: exceptions must derive from BaseException 其实是个低级错误,我个人认为是因为没有找到要运行的载体. 问 ...
报错解决：alueError: When using data tensors as input to a model, you should specify the `steps_per_epoch
报错解决:valueError: When using data tensors as input to a model, you should specify the steps_per_epoch ...
#报错解决#TypeError Failed to execute ‘readAsText‘ on ‘FileReader‘ parameter 1 is not of type ‘Blob‘
问题描述在实现需求:批量下载的时候,返回的数据类型设置为blob $http({method: 'post',url: dataurl,data: bodyParam,headers: { 'Con ...
Pytorch——报错解决：RuntimeError: Output 0 of SelectBackward is a view and is being modified inplace.
我在做Semi-Supervised过程中,需要分别计算labeled和unlabeled data的loss,但是在多卡分布式过程中,不允许出现同一个model前向两次的情况,会报错(Runtime ...
pytorch报错： invalid argument 0: Sizes of tensors must match except in dimension 0.
这个问题是在dataloader遇到的,网上一般是数据的维度不同,但我这里其实是最后一份数据和之前的数据比数量不足,所以凑不够一个batch,所以我的做法是加入drop_last=True意思是如果数 ...
报错解决：TypeError: Object type class 'str' cannot be passed to C code
此文首发于我的个人博客:报错解决 TypeError Object type class 'str' cannot be passed to C code - zhang0peter的个人博客下午在 ...
安装pytorch报错及解决
环境,win10,64位,Anaconda4.10.3,Cuda 9.2.148,cudnn应该为v7.4.2.24 其中在Anaconda中好像安装tensorflow-gpu 1.9 ,pytho ...
python3运行报错：TypeError: Object of type ‘type‘ is not JSON serializable解决方法（详细）
python3运行报错:TypeError: Object of type 'type' is not JSON serializable解决方法(详细) 参考文章: (1)python3运行报错:T ...
TypeError: ‘builtin_function_or_method‘ object is not subscriptable 报错解决方法
TypeError: 'builtin_function_or_method' object is not subscriptable 报错解决方法参考文章: (1)TypeError: 'buil ...

Pytorch——报错解决：TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, found

报错信息

解决方法

示例

Pytorch——报错解决：TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, found相关推荐

最新文章

热门文章