Saving and Loading Models (Coding TensorFlow)

Training models can take a very long time, and you definitely don’t want to have to retrain everything over a single mishap. Make sure you listen to Magnus as he explains the importance of saving and loading models so you won’t end up losing your valuable training time. Follow along in this episode of Coding TensorFlow with the Colab provided below. And as always, let us know what you think in the comments below!

[theRenjie]

I have been wondering how to save the model when training.... it's really annoying when you don't know the Patience to set in the earlystopping.. I learned a lot on checkpoint and model saving skills in this video. Thank you!

[frankfluit]

Thank you very very much, really helpful during the final stages of my thesis writing

[felipeguimaraes]

I was hoping that we would learn to save the models directly using Tensorflow, not Keras.

[jaffarbh]

Great. I would love to see more coverage on TensorFlow Serving next time. Thanks?

[CalogeroZarbo]

The two thumb down are PyTorch and Caffe. Great video by the way.

[moustafa_shomer]

you really helped me, thank you very much, u saved us time and a lot of work going through all that searching part, , deeply thankful
.

[aminuabdulsalami]

The drama is quite interesting and informative

[JZX]

Amazing content, needed to learn this

[ameytilve]

Where does model get saved ??on google drive?

[emanuelebarberis]

I'm finding a problem in the accuracy on the same testset between a model and the same model reloaded in a different python kernel. Someone know why?

[SantoshGupta-jnwn]

Is this the same Magnus from Hvass Laboratories
?

[ZOBAER]

Say we are going to save a trained CNN model. Can we save only the trained convolutional base part instead of saving the entire model including the fully connected part? The purpose is mainly to use the convolutional base as pretrained feature extractor and fine tune the fully-connected part. How can we do that?

[АртемСкачков-чл]

How save model MobileNet in .pb or . pbtxt format?

[Acer]

thank you. I can't access your code for Save and Restore Model.

[tiagocabral]

Ok, but what if I have to load the saved model on another notebook? I've tried this and both loss and accuracy from the trained model were not loaded.

[tallwaters]

This is great thank you, this is one of the things that always confused me about TF

[kothapallivijayakumar]

am getting ValueError: You are trying to load a weight file containing 16 layers into a model with 0 layers
while using load_model

[mtalhakhalid]

Which error chk didnu use DGE mae ??
Optimize your label add weights to it

[camus]

Ok, but how can i save and really use a model out of colab/jupyter ?
In case, on my PC .

[bryarmustafa]

Machine learning blippi! love you man.