Add notebook for fine-tuning YoloNAS with QAT (#1638)

Deci-AI · Nov 23, 2023 · 26d62e0 · 26d62e0
1 parent e98cec5
commit 26d62e0
Show file tree

Hide file tree

Showing 7 changed files with 3,696 additions and 5 deletions.
diff --git a/Makefile b/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)

diff --git a/README.md b/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