How to change checkpoint interval in Yolo 2 (darflow) python?

Question

I'm Training Yolo2 (darkflow) and want to change the interval (e.g. every 100 steps) to make checkpoints.

Train command:

python flow --model cfg/yolo.cfg --load bin/yolo.weights --train --annotation train/XML --dataset train/Images

Something like this:

--set ckpt 100

I could not find the command. Thank you !

Answer 1

You can try changing line 70 of file flow.py (https://github.com/thtrieu/darkflow/blob/5fe906b347ab8584354e375a14ea2e7bdf5188d7/darkflow/net/flow.py#L70)

ckpt = (i+1) % (self.FLAGS.save // self.FLAGS.batch)

to

  ckpt = (i+1) % 100