예제 #1
0
 def show_report(self):
     model_hook = ModelHook(self._model, self._input_size)
     leaf_modules = model_hook.retrieve_leaf_modules()
     stat_tree = convert_leaf_modules_to_stat_tree(leaf_modules)
     collected_nodes = stat_tree.get_collected_stat_nodes(
         self._query_granularity)
     report = report_format(collected_nodes)
     print(report)
예제 #2
0
 def show_report(self):
     collected_nodes = self._analyze_model()
     report = report_format(collected_nodes)
     print(report)