set_gpu_mode

Function Usage

Schedules AMCT weight quantization to the GPU.

Constraints

The GPU environment is available and CUDA10.0 is supported. This API does not support GPU card selection. You can select a GPU card by using the CUDA environment variable (CUDA_VISIBLE_DEVICES) or by using the set_device() API of PyCaffe.

Prototype

set_gpu_mode()

Command-line options

None

Returns

None

Outputs

None

Examples

1
2

import amct_caffe as amct
amct.set_gpu_mode()

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