hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

No way to pass epochs, LR etc as parameters #3

Closed mon28 closed 4 years ago

mon28 commented 4 years ago

So far there is no way to pass parameters to the train model while initiating a new training from scratch.

For now one can edit the following file from source. tensorflow-yolov4/py_src/yolov4/tf/__init__.py

hhk7734 commented 4 years ago

tf: train: move learningrate* to argumetn 03f8e36