set_cpu_mode

Product	Supported
Atlas 350 Accelerator Card	x
Atlas A3 training product/Atlas A3 inference product	x
Atlas A2 training product/Atlas A2 inference product	x
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Schedules AMCT weight quantization to the CPU.

1	set_cpu_mode()

None

None

If neither the set_gpu_mode API nor the set_cpu_mode API is called, AMCT weight quantization is scheduled to the CPU.

import amct_caffe as amct
amct.set_cpu_mode()

Parent topic: Common APIs