如何重新连接到 Google Colab 上正在进行的进程

Question

我最近开始使用 Google Colab 来训练我的 CNN 模型。训练一次总是需要大约 10 多个小时。但我不能在这 10 多个小时内呆在同一个地方，所以我总是关闭我的笔记本电脑，让这个过程继续进行。

我的代码会自动保存模型。我发现当我与 Colab 断开连接时，断开连接后该过程仍在保存模型。

以下是问题：

1. 当我尝试重新连接到 Colab 笔记本时，它总是卡在“INITIALIZAING”阶段并且无法连接。我确定该过程正在运行。我如何知道进程是否结束？

2. 有没有办法重新连接到正在进行的进程？在训练过程中观察训练损失对我来说会很好。

对不起，我的英语不好，非常感谢。

Answer 1

将丢失结果输出到驱动器中保存的日志文件中，并定期检查此文件。

你可以像这样运行你的训练过程：

!log_file = "/content/drive/My Drive/path/log.log"

!python train.py > "${log_file}"

Answer 2

1. 第一个问题：从运行时菜单重新启动运行时
2. 第二个问题：我认为你可以使用 tensorboard 来监控你的工作。

Answer 3

似乎没有正常的方法可以做到这一点。但是您可以使用当前训练纪元号将模型保存到 Google Drive，因此当您在 Google Drive 上看到类似“my_model_epoch_1000”的内容时，您就会知道该过程已经结束。