Skip to content

Draft: [Enhancement] QAT Upgrade

Benjamin Halimi requested to merge upgrade_qat into dev

Description

This merge request upgrade the code base of the QAT, that is (for now):

  • Add a routine that transforms the trained graph by replacing the LSQ Node with Quantizers, in order to use integer kernels.
  • Add back the forwardDims() in the LSQ Nodes (required)
  • Enhance the node insertion and in particular the BatchNorm insertion routine.

Merge request reports

Loading