wemindful
commented
2 years ago RuntimeError: shape '[4, 3, 6, 12, 64]' is invalid for input of size 589824
Closed wemindful closed 2 years ago
wemindful
commented
2 years ago RuntimeError: shape '[4, 3, 6, 12, 64]' is invalid for input of size 589824
let's test the original model first: Traceback (most recent call last): File "prune_yolov5s.py", line 43, in
origin_model_metric = eval_model(model)
File "prune_yolov5s.py", line 38, in
model=model)
File "/home/yolov5_prune-6/test.py", line 151, in test
data = parse_data_cfg(data)
File "/home/yolov5_prune-6/utils/parse_config.py", line 39, in parse_data_cfg
key, val = line.split('=')
ValueError: not enough values to unpack (expected 2, got 1)