nvflare.app_opt.pt.quantization.dequantizer module

class ModelDequantizer[source]

Bases: DXOFilter

Filter to dequantize Shareable object to recover from quantization

Parameters:

None

dequantization(params: dict, quant_state: dict, quantization_type: str, source_datatype: dict, fl_ctx: FLContext)[source]
process_dxo(dxo: DXO, shareable: Shareable, fl_ctx: FLContext) → None | DXO[source]

Filter process apply to the Shareable object.

Parameters:

  • dxo – data to be processed

  • shareable – that the dxo belongs to

  • fl_ctx – FLContext

Returns: DXO object with dequantized weights