Pruning was unsuccessful，HELP! about yolov5prune HOT 1 OPEN

Normal training and sparse training are possible, but an error is displayed during pruning. Does the code need to be modified? Please explain in detail how to fix it，THANKS

Traceback` (most recent call last):
File "prune.py", line 790, in
test_prune(opt.data,

File "prune.py", line 451, in test_prune
pruned_model = ModelPruned(maskbndict=maskbndict, cfg=pruned_yaml, ch=3).cuda()

File "/home/yolov5prune-main/models/yolo.py", line 250, in init
self.model, self.save, self.from_to_map = parse_pruned_model(self.maskbndict, deepcopy(self.yaml), ch=[ch]) # model, savelist

File "/home/yolov5prune-main/models/yolo.py", line 654, in parse_pruned_model
cv2out = int(maskbndict[named_m_cv2_bn].sum())

KeyError: 'model.2.cv2.bn'

git pull and try again.
'model.2.cv2.bn' is the cv2 module in the first C3 modules，it should be in the key of maskbndict.

from yolov5prune.