deep-diver / Soccer-Ball-Detection-YOLOv2

YOLOv2 trained against custom dataset
116 stars 43 forks source link

How to change the epoch value #13

Closed suravijayjilla closed 4 years ago

suravijayjilla commented 5 years ago

Hi,

I want to change my epoch value. But I didn't find any parameter related to this. Please help me on this.

deep-diver commented 4 years ago

check this out

options = {"model": "cfg/yolo_custom.cfg", "load": "bin/yolo.weights", "batch": 8, "epoch": 100, "gpu": 1.0, "train": True, "annotation": "./annotations/", "dataset": "./images/"}

suravijayjilla commented 4 years ago

hi,

Thanks for your valuable reply..