下载
中文
注册

Mask

用法

python3 mask_generate_model.py -token <max_token_cnt> -p <process_id> -pool <pool_size> -t <npu_type>

参数名称

<max_token_cnt>:算子生成token的最大值,默认为2500,建议设置范围为[1, 100000]。

<process_id>:批量生成算子多进程调度的进程ID,默认值为“0”,无需设置。

<pool_size>:算子生成进程池大小,默认为16。

<npu_type>:硬件形态,只支持“310P”

说明

无。

涉及接口

AscendIndexTS