文章分类 -  深度学习

摘要:转载:https://www.zhihu.com/question/319682377 一 pytorch模型保存与加载 模型包括模型结构和模型参数,因此,有不同的保存模型的方式。 1 官方推荐 1.1 只保存模型参数 优点:只保存模型参数,该方法速度快,占用的空间少 缺点:重载模型时,必须知道模型 阅读全文
posted @ 2022-06-17 18:22 IllidanStormrage 阅读(536) 评论(0) 推荐(1) 编辑
摘要:配置Docker环境首先要拉去镜像,可以利用生成的镜像直接拉取,也可自定义配置Dockerfile文件,生成镜像。 1. 直接拉取pytorch镜像 可以在Docker官网查找对应的pytorch镜像进行拉取。 如拉取最新版本的镜像 docker pull pytorch/pytorch:lates 阅读全文
posted @ 2022-06-08 18:39 IllidanStormrage 阅读(1358) 评论(0) 推荐(0) 编辑