search

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
Apache License 2.0
190 stars 46 forks source link

Getting KeyError issue model.2.cv2.weight #11

Closed muhammadabdullah34907 closed 4 years ago

muhammadabdullah34907 commented 4 years ago

Installed exact same versions of dependencies but getting this keyerror issue.

Traceback (most recent call last): File "main.py", line 208, in model = load_model() File "main.py", line 39, in load_model {k: v for k, v in ckpt['model'].state_dict().items() if model.state_dict()[k].numel() == v.numel()} File "main.py", line 39, in {k: v for k, v in ckpt['model'].state_dict().items() if model.state_dict()[k].numel() == v.numel()} KeyError: 'model.2.cv2.weight

muhammadabdullah34907 commented 4 years ago

fixed. this is because original .pt model is updated.