如何训练TF模型检查点到Google Drive上,同时在本地机器上使用它们进行推理?

时间:2019-03-29 14:29:47

标签: python tensorflow jupyter-notebook google-colaboratory

我已将我的google驱动器安装在可乐笔记本中,

from google.colab import drive
drive.mount('/content/drive')

现在我可以从衣领笔记本中看到/ content / drive /我的云端硬盘中的Google驱动器文件夹

在训练我的张量流模型时,我已在此路径中将检查点保存为:

save_path = saver.save(sess, "/content/drive/My Drive/Models/model.ckpt", global_step=e)

我也可以在我的Google云端硬盘中看到它们。 但是现在出现了两个问题:

1)如果我在加载模型时下载了这些模型检查点,以在本地木星笔记本中使用它们:

loader=tf.train.import_meta_graph(model_graph_path)

它抛出一个异常,说: OSError:文件/ content / drive /我的驱动器/模型/model.ckpt-700.meta不存在。

当我将其下载到本地计算机并提供本地副本的新路径时,为什么要在此路径中查找它?

2)如果我从Google驱动器方面更改了这些检查点的目录,它仍会在原始路径中查找它们,并抛出与以前相同的异常

我怎样才能将模型检查点保存在已安装的Google驱动器上,以备日后下载它们以在jupyter的本地计算机上恢复模型?

谢谢。

0 个答案:

没有答案