Hello,
I want to train the YOLOv5 model (YOLOv5 | PyTorch 13)  from scratch (not using the pretrained weights) on my own dataset and classes for a task of Face Mask Detection.
I have seen that in  order to train I should load:
model = torch.hub.load(‘ultralytics/yolov5’, ‘yolov5s’, autoshape=False, pretrained=False)  # load scratch
However, How do I actually train it? Can I use it as one layer in my model?
Thank you,
Almog