(model)
| 44 | return iter(params) |
| 45 | |
| 46 | def quant_parameters(model): |
| 47 | params = [] |
| 48 | for n, m in model.named_parameters(): |
| 49 | if n.find('scale') > -1 or n.find('zero_point') > -1: |
| 50 | params.append(m) |
| 51 | return iter(params) |
| 52 | |
| 53 | |
| 54 | def trainable_parameters(model): |