diff --git a/readme.md b/readme.md index 4268c74..cb536f0 100644 --- a/readme.md +++ b/readme.md @@ -50,7 +50,7 @@ NVIDIA DeepStream SDK 6.0 configuration for YOLO models * [NVIDIA DeepStream SDK 6.0](https://developer.nvidia.com/deepstream-sdk) * [DeepStream-Yolo](https://github.com/marcoslucianops/DeepStream-Yolo) -**For YOLOv5**: +**For YOLOv5 and YOLOR**: * [PyTorch >= 1.7.0](https://pytorch.org/get-started/locally/)