我想知道在每次训练后使用tf.train.Saver()保存模型时究竟保存了什么。与我习惯使用Keras模型的文件相比,该文件似乎有点大。现在,每次保存时,我的RNN占用900 MB。有什么办法告诉保护程序只保存可训练的参数?我还想要一种只保存部分模型的方法。我知道我可以获取我定义的变量并使用numpy格式保存它们,但是当我使用RNN类时,我无法直接访问它们的权重,我查看了代码,没有像get_weights这样的东西了。
| 归档时间: |
|
| 查看次数: |
2734 次 |
| 最近记录: |