CODEWITHSUNDEEP
pythontensorflow
megabolo
megabolo

Reputation: 31

How to retrain model in graph (.pb)?

I have model saved in graph (.pb file). But now the model is inaccurate and I would like to develop it. I have pictures of additional data to learn, but I don't if it's possible or if it's how to do it? The result must be the modified of new data pb graph.

Upvotes: 2

Views: 1937

Answers (2)

ReInvent_IO
ReInvent_IO

Reputation: 477

The mozzila/DeepSpeech community has contributed a way to initialize training from a frozen graph(.pb). It does not restores optimizer parameters, so adjusting the learning rate is necessary.

You could find the code at:

https://github.com/mozilla/DeepSpeech/blob/master/DeepSpeech.py#L1562

Hope this helps!

Upvotes: 1

AlienBuster
AlienBuster

Reputation: 73

It's a good question. Actually it would be nice, if someone could explain how to do this. But in addition i can say you, that it would come to "catastrophic forgetting", so it wouldn't work out. You had to train all your data again.

But anyway, i also would like to know that espacially for ssd, just for test reasons.

Upvotes: 3

Related Questions