sovit-123 fe3449a52d Updated Super Gradients version		1 gadu atpakaļ
..
readme_images	bf9b711387 Train YOLO NAS blog post	1 gadu atpakaļ
README.md	bf9b711387 Train YOLO NAS blog post	1 gadu atpakaļ
YOLO_NAS_Fine_Tuning.ipynb	fe3449a52d Updated Super Gradients version	1 gadu atpakaļ
YOLO_NAS_Large_Fine_Tuning.ipynb	ed6214fca7 YOLO NAS Large training notebook with inference	1 gadu atpakaļ
inference.ipynb	bf9b711387 Train YOLO NAS blog post	1 gadu atpakaļ

README

This folder contains the notebooks for training YOLO NAS models on custom dataset. This is part of the blog post - Train YOLO NAS on Custom Dataset.

There are three notebooks in this folder:

YOLO_NAS_Fine_Tuning.ipynb: For fine-tuning YOLO NAS Small, Medium, and Large models for 50 epochs.
YOLO_NAS_Large_Fine_Tining.ipynb: For fine-tuning the Large model for 100 epochs.
inference.ipynb: To carry out inference on the test set.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.