Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add notebook for fine-tuning YoloNAS with QAT #1638

Merged
merged 49 commits into from
Nov 23, 2023
Merged
Show file tree
Hide file tree
Changes from 39 commits
Commits
Show all changes
49 commits
Select commit Hold shift + click to select a range
10050dd
YoloNAS fine-tune with QAT (not working)
BloodAxe Nov 10, 2023
acccb5e
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 10, 2023
363f967
YoloNAS fine-tune with QAT (not working)
BloodAxe Nov 10, 2023
13ffddf
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 13, 2023
eeb45cb
Disable model averaging for QAT
BloodAxe Nov 14, 2023
1bebc7c
Change default fusion mode for QARepVGGBlock to partial fusion
BloodAxe Nov 14, 2023
c3b32e1
Replace assert with runtime check and use infer_model_device
BloodAxe Nov 14, 2023
35fa9f6
Replace assert with runtime check and use infer_model_device
BloodAxe Nov 14, 2023
8178c9d
Update notebook
BloodAxe Nov 14, 2023
8d033c2
Added notebook to NOTEBOOKS_TO_RUN
BloodAxe Nov 14, 2023
a11f354
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 14, 2023
01c219b
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 14, 2023
fc28ca9
Fix link
BloodAxe Nov 14, 2023
0dd746a
Revert
BloodAxe Nov 14, 2023
693c58f
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 15, 2023
dae4971
Added random_aspect_ratio to PPYoloECollateFN
BloodAxe Nov 17, 2023
6beecf0
Check whether additional samples is in dict
BloodAxe Nov 17, 2023
293a7ec
Fix bug that class_names was ignored in DetectionMetrics when include…
BloodAxe Nov 17, 2023
1c49ff0
DetectionAutoPadding._get_padding_params is not classmethod
BloodAxe Nov 17, 2023
ef3a4fd
Allow YoloNASStem to have stride argument
BloodAxe Nov 19, 2023
8e41508
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 20, 2023
9fb2061
Allow use of multi-scale training
BloodAxe Nov 20, 2023
c119c54
Allow use of multi-scale training
BloodAxe Nov 20, 2023
f687f8d
Allow use of multi-scale training
BloodAxe Nov 20, 2023
346fa8e
Add no grad & cast to float to support predictions in FP16 on CPU
BloodAxe Nov 20, 2023
e7794c0
Added logging
BloodAxe Nov 20, 2023
d74757a
Fix warning
BloodAxe Nov 20, 2023
cdba4aa
Fix warning
BloodAxe Nov 20, 2023
79917ad
Fix warning
BloodAxe Nov 20, 2023
35241b8
Fix warning
BloodAxe Nov 20, 2023
7a57c97
Fix warning
BloodAxe Nov 20, 2023
b2da5c0
Fix warning
BloodAxe Nov 20, 2023
e86ce5d
Fix warning
BloodAxe Nov 20, 2023
43b2615
Fix warning
BloodAxe Nov 20, 2023
f3f1b5b
Fix warning
BloodAxe Nov 20, 2023
8d6c618
Fix visualization
BloodAxe Nov 20, 2023
feac1cb
Fix visualization
BloodAxe Nov 20, 2023
a3d64c2
Merge branch 'feature/SG-000-improve-detection' into feature/SG-1214-…
BloodAxe Nov 21, 2023
0dabc78
Update notebook
BloodAxe Nov 21, 2023
3073d4b
Rollback file
BloodAxe Nov 21, 2023
2efcd81
Rollback file
BloodAxe Nov 21, 2023
59c1143
Rollback file
BloodAxe Nov 21, 2023
43283d3
Rollback files
BloodAxe Nov 21, 2023
87c2f86
Rollback files
BloodAxe Nov 21, 2023
be71bd8
Rollback files
BloodAxe Nov 21, 2023
ba2419f
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 21, 2023
b4dc002
Merge branch 'master' into feature/SG-1214-update-yolo-nas-notebook
BloodAxe Nov 23, 2023
6a4b8b9
Remove last line
BloodAxe Nov 23, 2023
596d330
Merge remote-tracking branch 'origin/feature/SG-1214-update-yolo-nas-…
BloodAxe Nov 23, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
Loading