Skip to content

Commit 53ce3af

Browse files
authored
Update torch.mdx
1 parent 6cd5655 commit 53ce3af

File tree

1 file changed

+3
-3
lines changed
  • i18n/zh/docusaurus-plugin-content-docs/current/backend-reference

1 file changed

+3
-3
lines changed

i18n/zh/docusaurus-plugin-content-docs/current/backend-reference/torch.mdx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ tensorrt推理引擎
133133

134134
| | 说明 | 备注 |
135135
|-------------------------|---------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------|
136-
| min/max | 模型最小/最大输入 | 形式上, 可以是`1`, `1x3x224x224`,或者 `1,1`(多输入网络)。 字典序排序;<br />`instance_num > 1`时,可以是以`;`隔开的多组配置 |
136+
| min/max | 模型最小/最大输入 | 形式上, 可以是`1`, `1x3x224x224`,或者 `1,1`(多输入网络)。 <br />`instance_num > 1`时,可以是以`;`隔开的多组配置 |
137137
| precision | 模型精度 | [fp32,fp16,int8,best]. 版本<=0.3.1b1时默认值为fp16,版本>0.3.1b1时默认为[fp16(SM>6.1), fp32(SM<=6.1)]; 如果精度不被支持,将自动退化到受支持的精度 |
138138
| precision::fp32 | 设置部分层的精度为fp32(覆盖precision的设置) | 层的名称(可以只提供名称的一部分),以逗号分开 |
139139
| precision::fp16 | 设置部分层的精度为fp16(覆盖precision的设置) | 层的名称(可以只提供名称的一部分),以逗号分开 |
@@ -158,8 +158,8 @@ tensorrt推理引擎
158158

159159
| | 说明 | 备注 |
160160
|-----------------------|----------------------------------------------------------------------------------------|----------|
161-
| TASK_DATA_KEY | 网络为单输入输出时,类型为at::Tensor/torch.Tensor, 网络为多输入输出时, 类型为vector/List. | 字典序排列 |
162-
| TASK_RESULT_KEY(输出) | 输出类型参照输入类型, postprocessor 可自定义输出 | 字典序排列 |
161+
| TASK_DATA_KEY | 网络为单输入输出时,类型为at::Tensor/torch.Tensor, 网络为多输入输出时, 类型为vector/List. | 字典序排列for trt<=9 |
162+
| TASK_RESULT_KEY(输出) | 输出类型参照输入类型, postprocessor 可自定义输出 | 字典序排列for trt<=9 |
163163

164164
### min()/max()
165165
输入范围将从tensorrt模型中读取。

0 commit comments

Comments
 (0)