Google Colab 保存和恢复模型(Pytorch)
COLA
🥤 COLA: Clean Object-oriented & Layered Architecture
项目地址:https://gitcode.com/gh_mirrors/col/COLA
免费下载资源
·
在Google Colab上,我们可以直接训练深度学习模型,接下来介绍一下Pytorch如何保存和恢复模型。
首先,将你的Google Drive挂载上。
-
点击Mount Drive
-
运行这段代码(上一步完成后,会自动生成的)
-
在文件夹中就能看到已挂载的Drive文件夹
保存模型
PATH = './drive/My Drive/tran.pth'
torch.save(best_model.state_dict(), PATH)
恢复模型(保证前面挂载了Google Drive的前提下,进行恢复)
PATH = './drive/My Drive/tran.pth'
checkpoint = torch.load(PATH)
model.load_state_dict(checkpoint)
问题:
- 为什么要把模型保存到Google Drive?
由于之前Google Drive运行一段时间,人不进行操作或网络不稳定,服务器会自动停止运行。保存到Google Drive更放心一些。
GitHub 加速计划 / col / COLA
11.69 K
3.04 K
下载
🥤 COLA: Clean Object-oriented & Layered Architecture
最近提交(Master分支:2 个月前 )
bda50471 - 5 个月前
65e9e39d
bump org.apache.maven.plugins:maven-compiler-plugin from 3.11.0 to 3.13.0 (#496)
bump org.apache.maven.plugins:maven-surefire-plugin from 3.1.2 to 3.2.5 (#497)
bump org.springframework.boot:spring-boot-dependencies from 3.1.0 to 3.3.0.(#498)
bump org.apache.maven.plugins:maven-javadoc-plugin from 3.6.0 to 3.7.0 (#499)
bump commons-cli:commons-cli from 1.5.0 to 1.8.0 (#500)
bump org.jacoco:jacoco-maven-plugin from 0.8.10 to 0.8.12 (#501)
bump org.apache.maven.plugins:maven-source-plugin from 3.2.1 to 3.3.1 (#502)
bump org.wiremock:wiremock-standalone from 3.0.1 to 3.0.3 (#503)
5 个月前
更多推荐
已为社区贡献4条内容
所有评论(0)