Flat

Usage

python3 flat_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.

<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 for distance calculation. You need to modify the parameters in the command.

Restrictions

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