search

THU-MIG / torch-model-compression

针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库
MIT License
244 stars 40 forks source link

self.index_mapping 代表什么意思？它是用来作什么的？ #12

Open scuizhibin opened 2 years ago

gdh1995 commented 2 years ago

呃大概是：

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

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