google-colaboratory - 神经网络记忆

Question

使用 Google Colab，我使用 tensorflow 制作了一个神经网络，它根据示例生成文本。我跑了 60 个 epoch。我怎样才能让我的神经网络保持它学到的东西。每当我重新运行它时，它都会重新开始。

标签： google-colaboratory

Answer 1

尝试在训练结束时保存模型，如下所示：

import tensorflow as tf
tf.keras.models.save_model(model, 'model/my_model')

然后你可以像这样加载模型：

tf.keras.models.load_model('model/my_model')