megengine.quantization.enable_fake_quant

enable_fake_quant(module)[source]

Recursively enable module fake quantization in QATModule through apply

Parameters

module (Module) – root module to do enable fake quantization recursively.