| 331 | |
| 332 | |
| 333 | class QuantizationSetting(): |
| 334 | def __init__(self) -> None: |
| 335 | # 子图切分与调度算法,可从 'pointwise', 'conservative', 'pursus', 'allin', 'pplnn' 中五选一,不区分大小写 |
| 336 | self.dispatcher = 'conservative' |
| 337 | |
| 338 | self.graph_format_setting = GraphFormatSetting() |
| 339 | |
| 340 | # ssd with loss check equalization 相关设置 |
| 341 | # may take longer time(about 30min for default 3 iterations), but guarantee better result than baseline |
| 342 | # should not be followed by a plain equalization when turned on |
| 343 | self.ssd_equalization = False |
| 344 | self.ssd_setting = SSDEqualizationSetting() |
| 345 | |
| 346 | # layer wise equalizition 与相关配置 |
| 347 | self.equalization = False |
| 348 | self.equalization_setting = EqualizationSetting() |
| 349 | |
| 350 | self.weight_split = False |
| 351 | self.weight_split_setting = WeightSplitSetting() |
| 352 | |
| 353 | # OCS channel split configuration |
| 354 | self.channel_split = False |
| 355 | self.channel_split_setting = ChannelSplitSetting() |
| 356 | |
| 357 | # Matrix Factorization Split. (Experimental method) |
| 358 | self.matrix_factorization = False |
| 359 | self.matrix_factorization_setting = MatrixFactorizationSetting() |
| 360 | |
| 361 | # activation 量化与相关配置 |
| 362 | self.quantize_activation = True |
| 363 | self.quantize_activation_setting = ActivationQuantizationSetting() |
| 364 | |
| 365 | # 参数量化与相关配置 |
| 366 | self.quantize_parameter = True |
| 367 | self.quantize_parameter_setting = ParameterQuantizationSetting() |
| 368 | |
| 369 | # 是否执行网络微调 |
| 370 | self.lsq_optimization = False |
| 371 | self.lsq_optimization_setting = LSQSetting() |
| 372 | |
| 373 | |
| 374 | self.blockwise_reconstruction = False |
| 375 | self.blockwise_reconstruction_setting = BlockwiseReconstructionSetting() |
| 376 | |
| 377 | # 是否启动 bias correction pass |
| 378 | self.bias_correct = False |
| 379 | self.bias_correct_setting = BiasCorrectionSetting() |
| 380 | |
| 381 | # 量化融合相关配置 |
| 382 | self.fusion = True |
| 383 | self.fusion_setting = QuantizationFusionSetting() |
| 384 | |
| 385 | # extension setting 只是一个空白的占位符,用来向你展示如何创建一个自定义的 setting 并传递参数。 |
| 386 | # extension setting shows you how to create a setting and pass parameter to passes. |
| 387 | # see ppq.quantization.quantizer.base.py |
| 388 | self.extension = False |
| 389 | self.extension_setting = TemplateSetting() |
| 390 |
no outgoing calls
no test coverage detected