我正在Google Colab(Tensorflow
)上运行goo.gl/8f3BQS
代码。代码运行良好,我可以对其进行正确测试。但是,当我尝试保存模型(以便以后下载)时,出现以下错误:
When eager execution is enabled, var_list must specify a list or dict of variables to save
我用来保存模型的代码是:
saver = tf.train.Saver()
save_path = saver.save(session, "./checkpoint.ckpt")
print('done saving at',save_path)
然后,我将使用以下代码下载检查点:
from google.colab import files
files.download(<filename>)
如何将模型保存到根文件夹然后下载?
谢谢。