Google Colab:从本地上传Pytorch模型后,“无法连接到运行时”

弗朗西斯

我正在使用一种简单(不一定高效)的方法来保存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断开连接,并且单击RECONNECTtab总是显示ALLOCATING-> CONNECTING(失败,在左下角显示“无法连接到运行时”消息)-> RECONNECT同时,执行任何一个单元将显示错误消息“无法执行单元,无法将执行消息发送到运行时:[object CloseEvent]”

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

为什么会发生?似乎已执行最后一行的Google帐户无法再连接到运行时。

编辑:

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

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

files.upload()仅用于上传文件。我们没有理由期望它返回一些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。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

使用Google Colab上传本地文件

当更改运行时类型选项未显示时,如何更改Google Colab中的运行时类型?

是否可以将vscode(在本地计算机上)与Google Colab(免费服务)运行时连接?

Google Colab中的本地运行时和托管运行时有什么区别?

在运行时本地化Google登录按钮

google.colab模块中是否有功能来关闭运行时

如何在Google Colab上删除本地上传的文件?

Google Colab-如何使用代码重新启动运行时?

连接到Google BQ时通过Spark-Submit运行时出错

从python3.7运行时连接到Google App Engine共享的内存缓存

设置更高的共享内存以避免Google Colab上的PyTorch运行时错误

Google Colab:本地运行时使用

无法连接到X服务器GOOGLE COLAB

无法连接到任何X显示器:Google Colab运行时

无法从Google Colab保存Tensorflow模型

如何以编程方式检查与哪个运行时Google Colab笔记本相连?

更新后,Google Colab将无法连接到本地运行时

如何使用Google Colab本地运行时并连接到Google云端硬盘

使用Google Colab上的Python Flask连接到本地主机吗?

App运行时Google Flutter Google Map API问题

突然无法在Google Colab中更改运行时的Python版本

将Google colab连接到Google Compute Engine实例上的运行时

Google Colab无法识别的运行时

无法通过自定义运行时在Docker中运行.NET Core应用的App Engine连接到Google Cloud SQL中的Postgres

无法连接到本地Google Cloud Datastore模拟器

Google Colab Pro错误:无法识别的运行时“ tensorflow”;默认为“ python3”笔记本设置

无法在Google Colab中加载spaCy模型

Google Chrome无法连接到本地主机

Google Cloud Storage 运行时错误

TOP 榜单

热门标签

归档