Add notebook for fine-tuning YoloNAS with QAT

Makefile

@@ -41,6 +41,7 @@ NOTEBOOKS_TO_RUN += notebooks/PTQ_and_QAT_for_classification.ipynb
 NOTEBOOKS_TO_RUN += notebooks/quickstart_segmentation.ipynb
 NOTEBOOKS_TO_RUN += notebooks/segmentation_connect_custom_dataset.ipynb
 NOTEBOOKS_TO_RUN += notebooks/transfer_learning_semantic_segmentation.ipynb
+NOTEBOOKS_TO_RUN += notebooks/yolo_nas_custom_dataset_fine_tuning_with_qat.ipynb
 
 # If there are additional notebooks that must not be executed, but still should be checked for version match, add them here
 NOTEBOOKS_TO_CHECK := $(NOTEBOOKS_TO_RUN)

README.md

@@ -225,6 +225,7 @@ model = models.get("model-name", pretrained_weights="pretrained-model-name")
 
 * [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://bit.ly/3SkMohx) [Object Detection Transfer Learning](https://bit.ly/3SkMohx)
 * [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/detection_how_to_connect_custom_dataset.ipynb) [How to Connect Custom Dataset](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/detection_how_to_connect_custom_dataset.ipynb)
+* [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/yolo_nas_custom_dataset_fine_tuning_with_qat.ipynb) [Quantization Aware Training YoloNAS on Custom Dataset](https://colab.research.google.com/github/Deci-AI/super-gradients/blob/master/notebooks/yolo_nas_custom_dataset_fine_tuning_with_qat.ipynb)
 
 ### How to Predict Using Pre-trained Model