我为我的分类任务使用了广泛而深入的学习模型。我已经在训练数据集上对它进行了训练,并希望将其保存,以便我以后可以将其用于预测目的。我尝试过使用:
saver = tf.train.Saver()
sess = tf.Session()
sess.run(tf.global_variables_initializer())
saver.save(sess, 'wide_n_deep')
但我得到的只是:
No variables to save
有人可以在这里建议我做错了吗?提前谢谢。