set_cpu_mode

产品支持情况

功能说明

调用该接口之后，AMCT执行权重量化的时候，使用CPU进行计算。

函数原型

1

set_cpu_mode()

参数说明

无

返回值说明

无

约束说明

如果既不调用set_gpu_mode接口也不调用set_cpu_mode接口。AMCT默认使用CPU进行权重的量化。

调用示例

1
2

import amct_caffe as amct
amct.set_cpu_mode()