[Fix] 'replace the usage of keys of self.io_module '

ModelTC / MQBench

Model Quantization Benchmark

Apache License 2.0

760 stars 137 forks source link

[Fix] 'replace the usage of keys of self.io_module ' #237

Open Menace-Dragon opened 1 year ago

Menace-Dragon commented 1 year ago

Original self.io_module is a dict composed of {'node. target': node}. When quantize model has shared convolution layer such as shared detector head, the 'node. target' will matches many nodes. However {'node. target': node} can not record all these situations. Here self.io_module is modified as {'node. target': [node1, node2, .....]} to fix the problem.

github-actions[bot] commented 1 year ago

This PR has not received any updates in 120 days. Please reply to this issue if this still unresolved!