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')
保存之後生成: