Loading Saved Model in Google Colab Using Yolov5

Community Help
1

hi, I did train the custom_data model in yolov5 using Google Colab Pro. And store the trained model using “Pickle” I need now to load it and work on it to detect the test images I suppose that should save me hours of training, How can i do that please

1 Like
2

Hi @mo_t1 - you’ll want to keep your model file as a .pt file. Here’s how to save and load model weights in Colab:

3

Hi explore this one and u’ll find an answer. Have a nice day! after take checkpoint u can continue from that checkpoint. And its saves to your cloud by using (MyDrive instead of Gdrive)

1 Like