TypeError: __init__() missing 1 required positional argument: 'dtype'

Traceback (most recent call last):
File "train.py", line 334, in
fit_one_epoch(net,yolo_loss,epoch,epoch_size,epoch_size_val,gen,gen_val,Unfreeze_Epoch,Cuda)
File "train.py", line 50, in fit_one_epoch
for iteration, batch in enumerate(gen):
File "D:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 521, in next
data = self._next_data()
File "D:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1183, in _next_data
return self._process_data(data)
File "D:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1229, in _process_data
data.reraise()
File "D:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\torch_utils.py", line 425, in reraise
raise self.exc_type(msg)
TypeError: init() missing 1 required positional argument: 'dtype'

在运行yolov4的时候到了冻结训练的时候出现了这个bug，请问是怎么回事，貌似是 gen =Dataloader()的问题，一直解决不了

budding的yolo4源码 睿智的目标检测30——Pytorch搭建YoloV4目标检测平台学习前言什么是YOLOV4代码下载YOLOV4改进的部分（不完全）YOLOV4结构解析1、主干特征提取网络Backbone2、特征金字塔3、YoloHead利用获得到的特征进行预测4、预测结果的解码5、在原图上进行绘制YOLOV4的训练1、YOLOV4的改进训练技巧a)、Mosaic数据增强b)、Label Smoothing平滑c)、CIOUd)、学习率余弦退火衰减2、loss组成a)、计算loss所需参数b)、y_pre是什么c)、y_tr https://blog.csdn.net/weixin_44791964/article/details/106214657?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522163054895116780255241095%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=163054895116780255241095&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_ecpm_v1~rank_v29_ecpm-2-106214657.first_rank_v2_pc_rank_v29&utm_term=pytorch%E6%90%AD%E5%BB%BAyolov4&spm=1018.2226.3001.4187