SQ8

Usage

python3 sq8_generate_model.py -d <dim> --cores <core_num> -p <process_id> -pool <pool_size>

Parameter

<dim>: feature vector dimension (D).

<core_num>: number of AI Cores of the Ascend AI Processor. The default value is 2.

<process_id>: ID of the process for batch generating operator multi-process scheduling tasks.

<pool_size>: size of the process pool for multi-process scheduling of operators generated in batches.

Description

Run the command to obtain a group of SQ8 operator model files for distance calculation. You need to modify the parameters in the command.

Restrictions

  • dim ∈ {64, 128, 256, 384, 512}
  • 0 ≤ pool_size ≤ 32