THU-MIG / torch-model-compression

针对pytorch模型的自动化模型结构分析和修改工具集,包含自动分析模型结构的模型压缩算法库

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

self.index_mapping 代表什么意思?它是用来作什么的?

scuizhibin opened this issue · comments

self.index_mapping 代表什么意思?它是用来作什么的?

呃大概是:

{
  "已知要剪枝的张量": {
    "被剪维度 | any": [
      ["受影响的张量1", "影响到的维度1"],
      ["受影响的张量2", "影响到的维度2"]
    ]
  }
}

影响到的维度也要跟着剪掉对应通道