IVFSQ8

Usage

python3 ivfsq8_generate_model.py -d <dim> -c <coarse centroid num> --cores <core_num> -p <process_id> -pool <pool_size>

Parameter

<dim>: feature vector dimension (D).

<coarse centroid num>: number of L1 cluster centroids.

<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 operator model files. You need to modify the parameters in the command.

Restrictions

  • dim ∈ {64, 128, 256, 384, 512, 1024, 2048}
  • coarse centroid num ∈ {1024, 2048, 4096, 8192, 16384}
  • 0 ≤ pool_size ≤ 32