产品
是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件
x
Atlas 200I/500 A2 推理产品
√
Atlas 推理系列产品
Atlas 训练系列产品
调用该接口之后，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()