detectron2中报错：TypeError: ‘＞‘ not supported between instances of ‘NoneType‘ and ‘int‘

一、问题描述

如题，使用detecton2时，出现类型错误：
TypeError: ‘>’ not supported between instances of ‘NoneType’ and ‘int’

二、原因

prefetch_factor 的默认值为 None。

三、解决方法

prefetch_factor 的参数传递过程为： build_detection_train_loader -> build_batch_data_loader -> torchdata.DataLoader，从 /torch/utils/data/dataloader.py 中可以看出，torchdata.DataLoader的默认值为 2，因此将 build_detection_train_loader 中 prefetch_factor 的值默认为2。