在Google Colab上,我们可以直接训练深度学习模型,接下来介绍一下Pytorch如何保存和恢复模型。

首先,将你的Google Drive挂载上。

  1. 点击Mount Drive
    在这里插入图片描述

  2. 运行这段代码(上一步完成后,会自动生成的)
    在这里插入图片描述

  3. 在文件夹中就能看到已挂载的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)

问题:

  1. 为什么要把模型保存到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 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐