- 博客(1)
- 资源 (2)
- 收藏
- 关注
转载 TensorFlow保存、加载模型参数 | 原理描述及踩坑经验总结
TensorFlow保存、加载模型参数 | 原理描述及踩坑经验总结 写在前面我之前使用的LSTM计算单元是根据其前向传播的计算公式手动实现的,这两天想要和TensorFlow自带的tf.nn.rnn_cell.BasicLSTMCell()比较一下,看看哪个训练速度快一些。在使用tf.nn.rnn_cell.BasicLSTMCell()进行建模的时候,遇到了模型保存、加载的问题。查找了一些博主的经验,再加上自己摸索,在这里做个笔记,总结经验。其中关键要素有以下3点:1.需要保存哪些变量(t
2021-06-13 11:21:44 115159
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人