How to recober the training prameters after the break during training

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.23k stars 1.24k forks source link

How to recober the training prameters after the break during training #267

Open czc00125 opened 3 years ago

czc00125 commented 3 years ago

Hi, very thanks for your contribution of YOLO. I have encountered a problem when I tried to train the COCO dataset. That is, how to reload the weights file such as '.meta' '.checkpoint' '.ckpt' to continue the training. I have read the code in "train.py", flags.DEFINE_string('weights', './data/yolov4.weights', 'pretrained weights') I think it can only load the file which ends with '.weights', so can you give me some instructions for this issue, thank you.