Getting a Subnetwork - 2.5 English - UG1333

Vitis AI Optimizer User Guide (UG1333)

Document ID
UG1333
Release Date
2022-06-15
Version
2.5 English

Call get_static_subnet() to get a specific subnetwork. The static_subnet can be used for finetuning and doing quantization.

pareto_global = ofa_pruner.load_subnet_config('pareto_global.txt')
static_subnet, static_subnet_config, flops, params = \ ofa_pruner.get_static_subnet(ofa_model, pareto_global['240']['subnet_setting'])