Flat

python3 flat_generate_model.py -d <dim> --cores <core_num> -p <process_id> -pool <pool_size> -t <npu_type>

<dim>: feature vector dimension. The default value is 512.

<core_num>: number of AI Cores of the Ascend AI Processor. The default value is 2. You do not need to configure this parameter.

<process_id>: ID of the process for multi-process scheduling of operators generated in batches. The default value is 0, and you do not need to set this parameter.

<pool_size>: size of the process pool for multi-process scheduling of operators generated in batches. The default value is 10.

<npu_type>: hardware form.

• For the Atlas 200/300/500 inference product and Atlas inference product, run the npu-smi info command on the server where the AI processor is installed and then delete the last digit of Name. The obtained value is the value of npu_type.
• For the Atlas 800I A2 inference server, run the npu-smi info command on the server where the AI processor is installed. The value of Name is the value of npu_type.
• For the Atlas 800I A3 SuperPoD Server, run the npu-smi info -t board -i 0 -c 0 command to obtain the NPU Name information. 910_NPU Name is the value of npu_type.

--help | -h: help information.

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

• dim ∈ {32, 64, 128, 256, 384, 512, 768, 1024, 1408, 1536, 2048, 3072, 3584, 4096}
• 0 ≤ pool_size ≤ 32