Training a retail Product Detection model using YOLO NAS
I am training a model using YOLO NAS pytorch. It’s working fine but my kernel keeps dying at epoch20. So i am trying to resume training from epoch 21 by loading epoch number and other details from average checkpoint file(average_model.pth) but no matter what it always resumes training from epoch 0. PFB code details
————————————-
#below part deals with respective YOLO image,label path declarations