模型
模块: polygraphy.tools.args
- class ModelArgs(model_opt_required: bool | None = None, required_model_type: str | None = None, input_shapes_opt_name: str | None = None, input_shapes_opt_doc: str | None = None, guess_model_type_from_runners: bool | None = None)[source]
基类:
BaseArgs模型：模型
依赖于
RunnerSelectArgs：如果 guess_model_type_from_runners == True
- 参数:
model_opt_required (bool) – 模型参数是否为必需项。默认为 False。
required_model_type (str) – 所需的模型类型。对于可处理多种模型类型的工具，请使用
None值。如果提供，则会导致该工具仅支持一种模型类型，并禁用--model-type选项。默认为 None。input_shapes_opt_name (str) – 用于指定输入形状的选项名称。将自动添加第二个选项名称，方法是删除指定选项中的最后一个
s并添加后缀-shapes。例如，值 “inputs” 将生成一个别名 “–input-shapes”。默认为 “inputs”。使用值False禁用该选项。input_shapes_opt_doc (str) – 输入形状选项的自定义帮助文本输出。
guess_model_type_from_runners (bool) – 是否基于已指定的运行器（如果有）来猜测模型类型。默认为 False。