tf的保存和讀取

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    # 加載訓練集和驗證集
    img, img_label = load_satetile_image(batch_size, dataset='train')
    img_valid, img_valid_label = load_satetile_image(batch_size, dataset='test')

    # 讀取模型
    saver = tf.train.Saver()
    saver.restore(sess, tf.train.latest_checkpoint('./model/'))

    for i in range(100):
	# 訓練


    # 保存模型
    saver.save(sess, './model/testmodel')

保存之後生成:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章