下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

概览

本节给出命令行场景所使用的参数概览。

  • 如果通过amct_caffe calibration --help命令查询出的参数未解释在表1,则说明该参数预留或适用于其他芯片版本,用户无需关注。
  • 使用amct_caffe命令行进行量化时,命令有两种方式,用户根据实际情况进行选择:
    • amct_caffe calibration param1=value1 param2=value2 ...(value值前面不能有空格,否则会导致截断,param取的value值为空)
    • amct_caffe calibration param1 value1 param2 value2 ...
表1 参数概览

参数项

参数说明

--help或--h

打印帮助信息。

--model

待量化网络的模型文件路径与文件名,格式为.prototxt。

--weights

待量化网络的权重文件路径与文件名,格式为.caffemodel。

--save_path

量化后模型的存放路径。

--input_shape

指定模型输入数据的shape。

若未使用--evaluator参数,则该参数必填。

--data_dir

与模型匹配的bin格式数据集路径。

若未使用--evaluator参数,则该参数必填。

--data_types

输入数据的类型。

若未使用--evaluator参数,则该参数必填。

--evaluator

基于"Evaluator"基类并且包含evaluator评估器的python脚本。

该参数与--input_shape--data_dir--data_types不能同时使用。

--calibration_config

训练后量化简易配置文件路径与文件名。

--batch_num

训练后量化推理阶段的batch数。

--gpu_id

指定使用GPU进行量化。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词