--dump_mode

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

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