set_cpu_mode

Description

Schedules AMCT weight quantization to the CPU.

Prototype

set_cpu_mode()

Command-Line Options

None

Return Value

None

Outputs

None.

Examples

1
2

import amct_caffe as amct
amct.set_cpu_mode()

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