TensorFlow2.0-Examples icon indicating copy to clipboard operation
TensorFlow2.0-Examples copied to clipboard

Train from the last checkpoint?

Open RegisOlivo opened this issue 5 years ago • 4 comments

generated files: checkpoint, yolov3.index, yolov3.data-00000-of-00001

Or just start train.py again?

RegisOlivo avatar Feb 05 '20 10:02 RegisOlivo

I have the same question,In Tensorflow1.x's version the Config's code are much different from version 2,How to deal with checkpoint?if I want to train from latest file?Thank you very much!

Fafa-DL avatar Feb 10 '20 03:02 Fafa-DL

I just changed the following in the train.py:

model = tf.keras.Model(input_tensor, output_tensors) #Already existing line
model.load_weights('./yolov3') #This is the new line

I don't know this for certain, but it seems like this just loads the saved weights and continues from there. I also see steady improvements using this method.

moritz157 avatar Aug 17 '20 21:08 moritz157

I just changed the following in the train.py:

model = tf.keras.Model(input_tensor, output_tensors) #Already existing line
model.load_weights('./yolov3') #This is the new line

I don't know this for certain, but it seems like this just loads the saved weights and continues from there. I also see steady improvements using this method.

Did that line do the trick? I am running into the same issue now and just want to confirm.

MhdMartini avatar Mar 31 '21 23:03 MhdMartini

I just changed the following in the train.py:

model = tf.keras.Model(input_tensor, output_tensors) #Already existing line
model.load_weights('./yolov3') #This is the new line

I don't know this for certain, but it seems like this just loads the saved weights and continues from there. I also see steady improvements using this method.

Did that line do the trick? I am running into the same issue now and just want to confirm.

As far as I remember, it did.

moritz157 avatar Apr 02 '21 12:04 moritz157