set_cpu_mode

Product	Supported
Atlas A3 training series products/Atlas A3 inference series products	x
Atlas A2 training products/Atlas A2 inference products	x
Atlas 200I/500 A2 inference product	√
Atlas inference series products	√
Atlas training products	√

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