Update readme (pytorch version recommendation)

Author: ersheng-ai

README.md

@@ -119,7 +119,10 @@ you can use darknet2pytorch to convert it yourself, or download my converted mod
 
 - **This script is to convert the official pretrained darknet model into ONNX**
 
-- **Pytorch version Recommended: 1.4.0 for TensorRT 7.0, 1.5.0 or higher for TensorRT 7.1**
+- **Pytorch version Recommended:**
+
+    - Pytorch 1.4.0 for TensorRT 7.0 and higher
+    - Pytorch 1.5.0 and 1.6.0 for TensorRT 7.1.2 and higher
 
 - **Install onnxruntime**
 
@@ -142,7 +145,10 @@ you can use darknet2pytorch to convert it yourself, or download my converted mod
 
 - **You can convert your trained pytorch model into ONNX using this script**
 
-- **Pytorch version Recommended: 1.4.0 for TensorRT 7.0, 1.5.0 or higher for TensorRT 7.1**
+- **Pytorch version Recommended:**
+
+    - Pytorch 1.4.0 for TensorRT 7.0 and higher
+    - Pytorch 1.5.0 and 1.6.0 for TensorRT 7.1.2 and higher
 
 - **Install onnxruntime**
 

tool/pytorch2onnx.py

@@ -23,7 +23,6 @@ def transform_to_onnx(cfgfile, weightfile, batch_size=1):
                       export_params=True,
                       opset_version=11,
                       do_constant_folding=True,
-                      # input_names=['input'], output_names=['output_1', 'output_2', 'output_3'],
                       dynamic_axes=None)
 
     print('Onnx model exporting done')