(model, requires_grad)
| 37 | return iter(params) |
| 38 | |
| 39 | def set_quant_parameters(model, requires_grad): |
| 40 | params = [] |
| 41 | for n, m in model.named_parameters(): |
| 42 | if n.find('scale') > -1 or n.find('zero_point') > -1: |
| 43 | m.requires_grad = requires_grad |
| 44 | return iter(params) |
| 45 | |
| 46 | def quant_parameters(model): |
| 47 | params = [] |