![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 50
pytorch
Zoran_卓
这个作者很懒,什么都没留下…
展开
-
一次成功安装pytorchGPU版本
昨天重装系统了,很多东西都重新安装过。下面介绍我安装pytorhGBP版本的过程,希望能帮到大家。1.安装cuda这个步骤可以参考我之前写的文章在windows10下安装tensorflow的GPU版本。安装cuda的内容是一致的。2.选择torch和torchvision的版本首先要清楚pytorch和cuda版本的对应关系再理清pytorch和torch版本对应关系例如我的python版本是3.7,cuda版本是10.0,那么我torch和torchvision的版本可以选择为1.2.0原创 2021-03-08 16:23:40 · 586 阅读 · 1 评论 -
pytorch笔记(二)——模型的保存与加载
1.保存和加载模型# 模型保存model = ModelClass(*args, **kwargs)torch.save(model, 'model.ckpt')# 模型加载model = torch.load(PATH) 保存整个神经网络的的结构信息和模型参数信息,save的对象是网络net。加载时则无需再定义网络。2.保存和加载模型参数# 模型参数保存model = ModelClass(*args, **kwargs)torch.save(model.state_dict(),原创 2021-01-04 13:53:05 · 696 阅读 · 3 评论 -
pytorch笔记(一)——tensor的storage()、stride()、storage_offset()
目录1 storage()2 stride()3 storage_offset()1 storage() pytorch中的一个tensor分为头信息区(Tensor)和存储区(Storage) 信息区主要保存着tensor的形状(size)、步长(stride)、数据类型(type)等信息。 而真正的数据则保存成连续数组,存储在存储区。 一般一个tensor都会有相对应的Storage,但也有另一种情况时多个tensor都对应着相同的一个Storage,这几个tensor只是头信息区不同原创 2020-12-06 23:20:03 · 9986 阅读 · 2 评论