andy-yun / pytorch-0.4-yolov3

Yet Another Implimentation of Pytroch 0.4.1 and YoloV3 on python3
MIT License
278 stars 72 forks source link

how to set maximum epoch on this code ? #85

Open Naveennavik opened 4 years ago

Naveennavik commented 4 years ago

This code keeps on running till we close the terminal Can we set the maximum epoch till which the training should run?

andy-yun commented 4 years ago

You can add max_epochs in [net] session.

Naveennavik commented 4 years ago

can u explain me in breif?

andy-yun commented 4 years ago

You try the following in config file [net] max_epochs=10000

Naveennavik commented 4 years ago

I train.py file or cfg.py file?

andy-yun commented 4 years ago

cfg/yolo_v3.cfg or similar config file.

Naveennavik commented 4 years ago

In yolo_v3.cfg u have mentioned steps=400000,450000 what is the meaning