Colaboratory上训练好的模型怎么保存

最新推荐文章于 2025-03-19 19:54:27 发布

ōu yǐng

最新推荐文章于 2025-03-19 19:54:27 发布

阅读量9.5k

点赞数 7

分类专栏：其他

本文链接：https://blog.youkuaiyun.com/oy_1024/article/details/105182216

版权

本文介绍了如何在Google Colaboratory上保存训练好的模型。通过连接和挂载Google Drive，用户可以在训练后将模型保存到云端，避免因Colab资源回收导致的模型丢失。详细步骤包括连接Google硬盘、挂载Drive，并给出了保存模型的示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Colab是一个能提供免费GPU/TPU算力的一个平台，我们可以使用其算力来加速程序，而每隔12个小时分配给我们的资源会强制收回，就需要我们学会上传、下载、保存我们训练好的模型，这里讲一下怎么保存训练好的模型。

连接Google硬盘

使用Google账号注册一个Google Drive的账号，注册成功就可以获得15G的云存储空间，这时我们只需要将数据和文件上传到Google Drive上，在Colab中就可以直接使用数据和文件

创建一个新的Colab文件，执行一下代码

!apt-get install -y -qq software-properties-common python-software-properties module-init-tools
!add-apt-repository -y ppa:alessandro-strada/ppa 2>&1 > /dev/null
!apt-get update -qq 2>&1 > /dev/null
!apt-get -y install -qq google-drive-ocamlfuse fuse
from google.colab import auth
auth.authenticate_user()
from oauth2client.client import GoogleCredentials
creds = GoogleCredentials.get_application_default()
import getpass
!google-drive-ocamlfuse -headless -id={creds.client_id} -secret={creds.client_secret} < /dev/null 2>&1 | grep URL
vcode = getpass.getpass()
!echo {vcode} |