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=/yolo_nas_s_soccer_players/ckpt_best.pth>, - num_classes=4) + checkpoint_path=/yolo_nas_s_soccer_players/ckpt_best.pth>, + num_classes=3) predictions = model.predict("messi_penalty.mp4") predictions.show(show_confidence=False) ```