--display_model_info

Applicability

Product	Supported
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

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.

Argument

Argument:

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

Restrictions: This option does not support displaying information about an offline model converted from a single-operator description file. It is mutually exclusive with --singleop.

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