Load ONNX model via torch.fx

Assuming a setting:

  1. I have a model.onnx
  2. I want to quant this model via a new quant algrithom(i.e. 4bit)

The question: I want to use the torch.fx as IR to represent the model.onnx, has any API? or any methods?

1 Like