CODEWITHSUNDEEP
deep-learningpytorchtraining-datayolov5
Ali Eren
Ali Eren

Reputation: 65

What is validation set for yolov5 training and is it necessary?

I'm training a yolov5 model and I have a dataset but I couldn't understand the differences between validation and test data. Should I split my dataset to train/test/val or just train/test?

Upvotes: 1

Views: 6882

Answers (2)

Sarah Syed
Sarah Syed

Reputation: 1

In Yolov5 (and Yolov8), you have to allocate images to the validation set. Training does not start unless the model detects a path to the validation set. The validation set is used as a regularization technique to prevent overfitting.

Upvotes: 0

kibromhft
kibromhft

Reputation: 1057

The validation dataset provides an unbiased assessment of a fitted model on the training dataset while fitting the model's hyper-parameters (e.g. the number of hidden units - layers and layer widths - in a neural network). Validation datasets can be used for early stop regularization (stopping training when the error in the validation dataset increases, as this is a sign of overfitting to the training dataset)

Upvotes: 1

Related Questions