Google Colab："Unable to connect to the runtime" 从本地上传 Pytorch 模型后

Question

我正在使用一种简单（不一定有效）的方法来保存 Pytorch 模型。

import torch
from google.colab import files

torch.save(model, filename) # save a trained model on the VM
files.download(filename) # download the model to local

best_model = files.upload() # select the model just downloaded
best_model[filename] # access the model

Colab 在执行最后一行时断开连接，点击 RECONNECT 选项卡总是显示 ALLOCATING -> CONNECTING（失败，左下角显示 "unable to connect to the runtime" 消息角）-> RECONNECT。同时，执行任何一个单元格都会给出错误消息 "Failed to execute cell, Could not send execute message to runtime: [object CloseEvent]"

我知道它与最后一行有关，因为我可以成功连接到我的其他 google 不执行该操作的帐户。

为什么会这样？似乎执行了最后一行的 google 个帐户无法再连接到运行时。

编辑：

一晚后，我可以在会话到期后重新连接到 google 帐户。我只是尝试了评论中的方法，发现只有 files.upload() Pytorch 模型会导致问题。上传完成后，Colab 会断开连接。

Answer 1

（我在阅读您的更新之前写了这个答案。认为它可能有帮助。）

files.upload() 仅用于上传文件。我们没有理由期望它 return 一些 pytorch type/model。

当你调用a = files.upload()时，a是一个文件名字典——一个大字节数组。

{'my_image.png': b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR....' }
type(a['my_image.png'])

就像你做的那样open('my_image', 'b').read()

所以，我认为下一行 best_model[filename] 尝试打印整个巨大的字节数组，这会使 colab 出错。

Answer 2

尝试禁用您的广告拦截器。为我工作

Google Colab："Unable to connect to the runtime" 从本地上传 Pytorch 模型后

Google Colab: "Unable to connect to the runtime" after uploading Pytorch model from local

data-persistence

pytorch

google-colaboratory