文档
注册

IVFSQT

为了减少train和add和耗时,需要生成FlatAT算子。其中,<dim>需与前者<dim_in>相同,<code_num>与<coarse_centroid_num>一致。

IVFast方案同IVFSQT。

用法

python3 ivfsqt_generate_model.py --cores <core_num> -d <dim_in> -r <compress_ratio> -c <coarse_centroid_num> -p <process_id>

参数名称

<dim_in>:输入特征向量维度。

<compress_ratio>:输入与输出维度的比值。

<coarse_centroid_num>:L1簇聚类中心个数。

<core_num>:昇腾AI处理器AI Core的个数,默认为“2”

<process_id>:批量生成算子多进程调度的进程ID。

说明

执行此命令,用户可以得到一组算子模型文件。

约束说明

  • dimIn ∈ {64, 128, 256, 384, 512, 1024}
  • dimOut ∈ {32, 64, 96, 128}
  • coarse_centroid_num ∈ {1024, 2048, 4096, 8192, 16384, 32768}
  • <dim_in>可以被<compress_ratio>整除。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词