set_cpu_mode
Description
Schedules AMCT weight quantization to the CPU.
Prototype
set_cpu_mode()
Parameters
None
Returns
None
Outputs
None
Example
import amct_caffe as amct amct.set_cpu_mode()
If neither the set_gpu_mode interface nor the set_cpu_mode API is called, AMCT weight quantization is scheduled to the CPU.