“启用急切执行后,`var_list`必须指定要保存的变量的列表或字典”错误

时间:2018-08-10 16:01:59

标签: python tensorflow keras google-colaboratory

我正在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>)

如何将模型保存到根文件夹然后下载?

谢谢。

0 个答案:

没有答案