coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本,以适应不同的需求。
793 stars 214 forks source link

稀疏化训练完后并没有生成 cfg 文件,直接剪枝报错 #103

Closed lguopeng closed 3 years ago

lguopeng commented 3 years ago

其中 prune_tiny_yolo.py 文件的 opt 参数如下 model_def = "cfg/yolov3-tiny.cfg" data_config = "cfg/oxfordhand.data" model = 'weights/last.pt'

报错: The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "prune_tiny_yolo.py", line 38, in origin_model_metric = eval_model(model) File "prune_tiny_yolo.py", line 33, in eval_model = lambda model:test(model=model,cfg=opt.model_def, data=opt.data_config) File "E:\python\YOLOv3-complete-pruning-master\test.py", line 69, in test for batch_i, (imgs, targets, paths, shapes) in enumerate(tqdm(dataloader, desc=s)): File "D:\programs\Python38\lib\site-packages\tqdm\std.py", line 1167, in iter for obj in iterable: File "D:\programs\Python38\lib\site-packages\torch\utils\data\dataloader.py", line 435, in next data = self._next_data() File "D:\programs\Python38\lib\site-packages\torch\utils\data\dataloader.py", line 1068, in _next_data idx, data = self._get_data() File "D:\programs\Python38\lib\site-packages\torch\utils\data\dataloader.py", line 1034, in _get_data success, data = self._try_get_data() File "D:\programs\Python38\lib\site-packages\torch\utils\data\dataloader.py", line 885, in _try_get_data raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) from e RuntimeError: DataLoader worker (pid(s) 8948, 244, 10520, 3460) exited unexpectedly

lguopeng commented 3 years ago

1