徐土豆
认证:优质创作者
作者动态
指令微调BLIP:一种对指令微调敏感的Q-Former设计
2天前
Kosmos-2: 在多模态大语言模型中引入基准和指代能力
3天前
Kosmos-1: 通用接口架构下的多模态大语言模型
4天前
ERNIE VIL 2.0,多模态模型的一种多视角预训练范式
5天前
再论系统复杂度控制:错误控制与复盘
6天前

TensorlFlow中的一些坑

model saver

  1. issue link:Saving and Restoring a trained LSTM in Tensor Flow问题描述:在保存TensorFlow中的RNN/LSTM模型的时候,需要在LSTM模型建立之后再定义saver如:
### Model Training and Saving code
### define the LSTM model code here
saver = tf.train.Saver()
### train process here
saver.saver(sess, saver_path) 
### Model Predict and Restore code
### define the LSTM model code here
saver = tf.train.Saver()
saver.restore(sess. saver_path)

事实上,如果是保存一般的variables也需要在定义了variable,如tf.get_variable()或者tf.Variable()中定义了之后,才能create Saver不然的话也会保存失败。下面就是一个例子

saver = tf.train.Saver()
    tf.Variable([1], tf.float32)
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())
        saver.save(sess, path)

这样运行将会报错:

ValueError: No variables to save

而将Saver的定义位置换一下,就可以解决问题了。

tf.Variable([1], tf.float32)
    saver = tf.train.Saver()
    with tf.Session() as sess:
        sess.run(tf.global_variables_initializer())
        saver.save(sess, path)
声明:本内容为作者独立观点,不代表电子星球立场。未经允许不得转载。授权事宜与稿件投诉,请联系:editor@netbroad.com
觉得内容不错的朋友,别忘了一键三连哦!
赞 1
收藏 2
关注 49
成为作者 赚取收益
全部留言
0/200
成为第一个和作者交流的人吧