aimet_onnx.apply_seq_mse

Top level APIs

aimet_onnx.apply_seq_mse(sim, inputs, num_candidates=20, nodes_to_exclude=None)[source]

Sequentially optimizes the QuantizationSimModel’s weight encodings to reduce MSE loss at layer outputs.

Parameters:
  • sim (QuantizationSimModel) – QuantizationSimModel instance to optimize

  • inputs (Collection[Dict[str, np.ndarray]]) – The set of input samples to use during optimization

  • num_candidates (int) – Number of encoding candidates to sweep for each weight. Decreasing this can reduce runtime but may lead to lower accuracy.

  • nodes_to_exclude (Optional[List[str]]) – List of supported node name(s) to exclude from sequential MSE optimization