--display_model_info

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.

See Also

None

Argument

Argument:

• 0 (default): disabled.
• 1: 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

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 model definition (ModelDef), weight, TBE kernels (tbekernels), task info (taskinfo), 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   ============

Applicability

Atlas 200/300/500 Inference Product

Atlas Training Series Product

Dependencies and Restrictions

None