【发布时间】:2021-09-29 21:54:13
【问题描述】:
执行命令后,在 Google Collab 笔记本中:
%load_ext tensorboard
%tensorboard --logdir <data_directory>
我无法将模型保存在云中,不仅自动保存,也无法手动保存。
有人知道为什么会发生这种情况?
【问题讨论】:
-
我不明白您所说的无法保存在云中的意思。您可以使用
ModelCheckpoints将模型保存在当前会话中,然后将训练后的输出文件发送到 can。 -
请向我们提供您的代码,您如何编写日志以及如何创建日志文件。此外,除了
ModelCheckpoints,您也可以按照您所说的“手动”进行操作,例如:创建一个writer = SummaryWriter(log_dir)并在每个需要的步骤中写入writer.add_scalar('my_val', scalar_value=my_val, global_step=my_step) -
@ans 感谢两位的回复。仅通过提供我在问题中提到的两行代码,一旦将 TensorBoard 模块加载到 Google Colab 中的虚拟机,它将不再保存文件,但是,它允许我编辑文件并运行它但不能保存它,这是我真正不明白的。我认为这要么是一个错误,要么我可能遇到权限问题或类似问题。我会看看 ModelCheckpoints 看看它是否有效,再次感谢
标签: tensorflow machine-learning save google-colaboratory tensorboard