Load megengine dumped model and visualize graph structure with tensorboard log files.
Load megengine dumped model and visualize graph structure with tensorboard log files.
...
@@ -48,10 +52,14 @@ def visualize(
...
@@ -48,10 +52,14 @@ def visualize(
:param model_path: dir path for megengine dumped model.
:param model_path: dir path for megengine dumped model.
:param log_path: dir path for tensorboard graph log.
:param log_path: dir path for tensorboard graph log.
:param input: user defined input data for running model and calculating stats, alternative with inp_dict, used when the model has only one input.
:param inp_dict: input dict for running model and calculating stats, alternative with input, used when the model has more than one input. When both input and inp_dict are None, a random input will be used.
:param cal_params: whether calculate and record params size.
:param cal_flops: whether calculate and record op flops.
:param cal_activations: whether calculate and record op activations.
:param logging_to_stdout: whether print all calculated statistic details.
:param bar_length_max: size of bar indicating max flops or parameter size in net stats.
:param bar_length_max: size of bar indicating max flops or parameter size in net stats.
:param log_params: whether print and record params size.
:param log_flops: whether print and record op flops.
:param log_activations: whether print and record op activations.