This release includes the following main new features.
-
FLOPs and parameter calculations. FLOPs and parameters could be computed via
oto.compute_flops()
andoto.compute_num_params()
. Detailed usages have been presented in sanity checks. -
Marking node groups as unprunable. Node groups could be marked as unprunable via either
node_ids
in topological view orparam_names
by DNN parameters, seeoto.mark_unprunable_by_node_ids
andoto.mark_unprunable_by_param_names
Concrete examples could be found at yolov5 and DemoNetGroupConvCase1. -
Pruning dependency graph visualization could display param names and shapes. Use
oto.visualize(display_params=True)
to enable this feature to visualize more information for debugging if needed. An example could be found here versus the version rendered viaoto.visualize(display_params=False)
.