--display_model_info

Applicable Products

All processors

Description

Displays the information about key resources occupied by the model as well as the build and run environment on the screen for building an original framework model.

Arguments

Argument:

0 (default): query disabled.
1: query enabled.

Restrictions: This option cannot be used together with --singleop for viewing information when converting a single-operator description file to an offline model.

Suggestions and Benefits

None

Example

The following uses a TensorFlow model as an example:

atc --model=$HOME/module/resnet50_tensorflow.pb  --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version> --display_model_info=1

After the command is executed successfully, you should see messages similar to the following:

============ Display Model Info start ============
# atc command used for model conversion
Original Atc command line: ${INSTALL_DIR}/bin/atc.bin --model=$HOME/module/resnet50_tensorflow.pb  --framework=3 --output=$HOME/module/out/tf_resnet50 --soc_version=<soc_version> --display_model_info=1
# ATC version, SoC version, and original framework type
system   info: atc_version[xxx], soc_version[xxx], framework_type[xxx].
# Memory usage, weight, number of logic streams, and number of events at run time
resource info: memory_size[xxx B], weight_size[xxx B], stream_num[xxx], event_num[xxx].
# Size of each partition in the offline model file, including the ModelDef, weight, tbe_kernels, task_info, and .so files
om       info: modeldef_size[xxx B], weight_data_size[xxx B], tbe_kernels_size[xxx B], cust_aicpu_kernel_store_size[xxx B], task_info_size[xxx B], so_store_size[xxx B].
============ Display Model Info end   ============

Dependencies and Restrictions

None

Parent topic: Debug Options