神经网络记忆

Question

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

Answer 1

尝试在训练结束时像这样保存模型：

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

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

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