Torch-Pruning pruner.step() 时报错 IndexError: list index out of range

pruner.step() 时报错 IndexError: list index out of range

Open byrantz opened this issue 1 year ago • 5 comments

试验时报错：

Traceback (most recent call last):
  File "print.py", line 113, in <module>
    pruner.step()
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/pruner/algorithms/metapruner.py", line 159, in step
    for group in self.prune_local():
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/pruner/algorithms/metapruner.py", line 201, in prune_local
    for group in self.DG.get_all_groups(ignored_layers=self.ignored_layers, root_module_types=self.root_module_types):
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/dependency.py", line 493, in get_all_groups
    group = self.get_pruning_group(
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/dependency.py", line 466, in get_pruning_group
    _fix_dependency_graph_non_recursive(*group[0])
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/dependency.py", line 450, in _fix_dependency_graph_non_recursive
    new_indices = mapping(new_indices)
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/_helpers.py", line 63, in __call__
    new_idxs = [
  File "/home/yxc/anaconda3/envs/top/lib/python3.8/site-packages/torch_pruning/_helpers.py", line 66, in <listcomp>
    if (i >= self.offset[0] and i < self.offset[1])
IndexError: list index out of range

请问这我模型的问题吗，torch-pruning 已经更新到最新