From fe5a0e7e9cbccd8f2b2a75162c4f4a785be06ed9 Mon Sep 17 00:00:00 2001
From: Pranoy Radhakrishnan <pranoyalkr@gmail.com>
Date: Sun, 13 Aug 2023 13:26:40 +0530
Subject: [PATCH] Doc changes (#1253)

* num classes specified was wrong

* wrong num_classes specified

---------

Co-authored-by: Ofri Masad <ofrimasad@users.noreply.github.com>
Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
---
 documentation/source/qat_ptq_yolo_nas.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/documentation/source/qat_ptq_yolo_nas.md b/documentation/source/qat_ptq_yolo_nas.md
index 419e9c533a..f5728f92d9 100644
--- a/documentation/source/qat_ptq_yolo_nas.md
+++ b/documentation/source/qat_ptq_yolo_nas.md
@@ -174,8 +174,8 @@ from super_gradients.common.object_names import Models
 from super_gradients.training import models
 
 model = models.get(Models.YOLO_NAS_S,
-                   checkpoint_path=<YOUR_CHECKPOINTS_ROOT_DIRECTORY>/yolo_nas_s_soccer_players/ckpt_best.pth>,
-                   num_classes=4)
+                  checkpoint_path=<YOUR_CHECKPOINTS_ROOT_DIRECTORY>/yolo_nas_s_soccer_players/ckpt_best.pth>,
+                  num_classes=3)
 predictions = model.predict("messi_penalty.mp4")
 predictions.show(show_confidence=False)
 ```