yolov5prune v7.0版本能用吗？

我用的是v5的7.0版本,稀疏训练的时候会报错，求解答报错如下：Traceback (most recent call last): File "C:\Users\Admin\Desktop\jianzhi\yolov5prune-6.0\train_sparity.py", line 674, in main(opt) File "C:\Users\Admin\Desktop\jianzhi\yolov5prune-6.0\train_sparity.py", line 571, in main train(opt.hyp, opt, device, callbacks) File "C:\Users\Admin\Desktop\jianzhi\yolov5prune-6.0\train_sparity.py", line 117, in train ckpt = torch.load(weights, map_location=device) # load checkpoint File "C:\Users\Admin.conda\envs\yolov5-7.0\lib\site-packages\torch\serialization.py", line 607, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "C:\Users\Admin.conda\envs\yolov5-7.0\lib\site-packages\torch\serialization.py", line 882, in _load result = unpickler.load() File "C:\Users\Admin.conda\envs\yolov5-7.0\lib\site-packages\torch\serialization.py", line 875, in find_class return super().find_class(mod_name, name) AttributeError: Can't get attribute 'DetectionModel' on <module 'models.yolo' from 'C:\Users\Admin\Desktop\jianzhi\yolov5prune-6.0\models\yolo.py'>