From 383e19b565a26d01df8214458bb4b00b37b11a1a Mon Sep 17 00:00:00 2001
From: Marcos Luciano <marcoslucianops@gmail.com>
Date: Sun, 9 May 2021 18:56:37 -0300
Subject: [PATCH] Update YOLOv5-4.0.md

---
 YOLOv5-4.0.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/YOLOv5-4.0.md b/YOLOv5-4.0.md
index f954a12..8580d75 100644
--- a/YOLOv5-4.0.md
+++ b/YOLOv5-4.0.md
@@ -36,7 +36,7 @@ pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
 sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
 git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
 cd torchvision
-export BUILD_VERSION=0.9.0  
+export BUILD_VERSION=0.9.0
 python3 setup.py install --user
 ```
 
@@ -109,7 +109,7 @@ cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.1/sources/yolo/yolov5s.eng
 
 Note: by default, yolov5 script generate model with batch size = 1 and FP16 mode.
 ```
-#define USE_FP32  // set USE_INT8 or USE_FP16 or USE_FP32
+#define USE_FP16  // set USE_INT8 or USE_FP16 or USE_FP32
 #define DEVICE 0  // GPU id
 #define NMS_THRESH 0.4
 #define CONF_THRESH 0.5