Update README.md

z1069614715 · Nov 28, 2022 · 0ec4126 · 0ec4126
1 parent 2e46a61
commit 0ec4126
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -315,10 +315,10 @@ image classifier implement in pytoch.
     default: False  
     导出tensorrt时是否显示日志.
   - **export**  
-    type: string, default: torchscript  choices: ['onnx', 'torchscript', 'tensorrt']
+    type: string, default: torchscript  choices: ['onnx', 'torchscript', 'tensorrt']  
     选择导出模型.
   - **device**  
-    type: string, default: torchscript  
+    type: string, default: ''  
     使用的设备.(cuda device, i.e. 0 or 0,1,2,3 or cpu)  
 
 <p id="ModelZoo"></p>