set_cpu_mode
功能说明
调用该接口之后,AMCT执行权重量化的时候,使用CPU进行计算。
函数原型
set_cpu_mode()
参数说明
无。
返回值说明
无。
函数输出
无。
调用示例
1 2 |
import amct_caffe as amct amct.set_cpu_mode() |
如果既不调用set_gpu_mode接口也不调用set_cpu_mode接口。AMCT默认使用CPU进行权重的量化。
父主题: 公共接口