Soccer-Ball-Detection-YOLOv2 icon indicating copy to clipboard operation
Soccer-Ball-Detection-YOLOv2 copied to clipboard

Changing checkpoint variable to custom value

Open getcontrol opened this issue 5 years ago • 1 comments

I have Darflow and your Jupyter notebook working great together. Thanks for this great resource!

I am working off an old Macbook Air with limited CPU/GPU so it trains very slowly. I've gotten up to batch 500 so far and then stopped the kernel to feed the checkpoint into the proceeding steps.

I understand that the lines:

options = {"model": "cfg/yolo_custom.cfg",
           "load": -1,
           "gpu": 1.0}

will utilize my last checkpoint (i.e. 500). Confirmed with output :

Loading from ./ckpt/yolo_custom-500

How do I configure the variable "load": -1, to jump to the provided pretrained weight i.e. 2300?

Thanks again for your great documentation !

getcontrol avatar Nov 14 '19 19:11 getcontrol

Hey @getcontrol,

All the checkpoints should be present in the ckpt folder. So, the moment, darkflow manages to save the checkpoint after finishing 2300 steps you will be able to access it. Did this answer your question?

Tanmay-Kulkarni101 avatar Nov 15 '19 16:11 Tanmay-Kulkarni101