昇腾社区首页
中文
注册

--framework

功能说明

原始框架类型。

关联参数

无。

参数取值

参数值:

  • 0:Caffe
  • 3:TensorFlow
  • 5:ONNX

参数值约束:必选参数,当前仅支持配置为0、3和5。

推荐配置及收益

无。

示例

  • Caffe框架:
    amct_acl --framework=0 --model=./pre_model/ResNet-50-deploy.prototxt --weight=./pre_model/ResNet-50-model.caffemodel --calibration_data=./image/calibration.bin --calibration_shape="data:32,3,224,224"  --output=./results/ResNet-50-model --soc_version=<soc_version> --log=info
  • TensorFlow框架:
    amct_acl --framework=3 --model=./pre_model/resnet_v1_50.pb --calibration_data=./image/calibration.bin --calibration_shape="input:16,224,224,3" --output=./results/resnet_v1_50 --soc_version=<soc_version> --log=info
  • ONNX网络模型:
    amct_acl --framework=5 --model=./pre_model/resnet101_v11.onnx --calibration_data=./images/calibration.bin --calibration_shape="input:16,3,224,224"  --out_nodes="Gemm_353:0" --output=./results/resnet101_v11 --soc_version=A<soc_version> --log=debug

依赖约束

无。