--dump_mode

Applicability

Product

Supported (Yes/No)

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

Atlas inference product

Atlas training product

Description

Sets whether to generate a JSON file with shape information.

See Also

Use this option in conjunction with --json, --mode = 1, --framework, and --om (as well as --weight if the original model is trained on Caffe).

Argument

  • 0 (default): Generates a JSON file without shape information.
  • 1: Generates a JSON file with shape information.

Suggestions and Benefits

None

Example

atc --mode=1 --om=$HOME/module/resnet50_tensorflow.pb --json=$HOME/module/out/tf_resnet50.json --framework=3 --dump_mode=1

Dependencies and Restrictions

None