Tensorflow: how to save/restore a model?

前端 未结 26 3218
迷失自我
迷失自我 2020-11-21 11:37

After you train a model in Tensorflow:

  1. How do you save the trained model?
  2. How do you later restore this saved model?
26条回答
  •  佛祖请我去吃肉
    2020-11-21 12:05

    For tensorflow-2.0

    it's very simple.

    import tensorflow as tf
    

    SAVE

    model.save("model_name")
    

    RESTORE

    model = tf.keras.models.load_model('model_name')
    

提交回复
热议问题