search

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs
https://arxiv.org/abs/2301.12900
MIT License
2.59k stars 318 forks source link

list index out of range #277

Open wujianfei5201314 opened 10 months ago

wujianfei5201314 commented 10 months ago

Traceback (most recent call last): File "D:\yolov81010\test1.py", line 440, in prune(args) File "D:\yolov81010\test1.py", line 369, in prune pruner = tp.pruner.GroupNormPruner( File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning\pruner\algorithms\group_norm_pruner.py", line 82, in init self._groups = list(self.DG.get_all_groups(root_module_types=self.root_module_types, ignored_layers=self.ignored_layers)) File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning\dependency.py", line 519, in get_all_groups group = self.get_pruning_group( File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning\dependency.py", line 477, in get_pruning_group _fix_dependency_graph_non_recursive(*group[0]) File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning\dependency.py", line 463, in _fix_dependency_graph_non_recursive new_indices = mapping(new_indices) File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning_helpers.py", line 91, in call new_idxs = [ _HybridIndex(idx=i.idx + self.offset[0], root_idx=i.root_idx) for i in idxs] File "D:\LenovoSoftstore\anaconda\envs\yolov8717\lib\site-packages\torch_pruning_helpers.py", line 91, in new_idxs = [ _HybridIndex(idx=i.idx + self.offset[0], root_idx=i.root_idx) for i in idxs] IndexError: list index out of range

Process finished with exit code 1

zmeng12 commented 4 months ago

请问有解决吗

Sarthak-22 commented 2 months ago

Hi, were you able to solve it ?