![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch入门-小土堆
文章平均质量分 84
我为什么可以这么菜
这个作者很懒,什么都没留下…
展开
-
5.模型训练
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P25~33 现有网络模型的使用及修改 官方文档 以VGG16为例 pretrained = True时是已经在ImageNet数据集上训练好的,需要先安装scipy库才能使用ImageNet数据集 ImageNet数据集 train_data = torchvision.datasets.ImageNet("./data_image_net", split='train', download=True, import torchv原创 2022-05-11 21:38:01 · 1416 阅读 · 0 评论 -
4.神经网络
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P16~24 nn.Module 神经网络的基本骨架,所有神经网络模块的基类。 官方文档 例子 import torch from torch import nn class Tudui(nn.Module): def __init__(self): # 这里是有操作的,code -> generate -> override methods -> --init__(self.Module) 可以快速重写这部原创 2022-05-11 16:52:07 · 461 阅读 · 0 评论 -
3.PyTorch加载数据
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P6~15 Dataset类 Dataset类一个表示数据集的抽象类torch.utils.data.Dataset 通过继承Dataset类可以实现数据集的加载,定义三个函数,分别是:初始化 __init__()、获得每一个数据 __getitem__()、数据长度 __len__()。 这里面的过程,要很清楚: __init__()用于初始化数据集的配置,例如数据集的路径、文件名称; __getitem__()用于获取给定键对应的数据原创 2022-05-10 23:11:35 · 798 阅读 · 0 评论 -
2. Python编辑器的选择、安装及配置(PyCharm、Jupyter安装)
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P2~5 PyCharm安装 pyCharm官网 这里安装的是专业版 其他一路默认安装即可 PyCharm配置 创建好项目后在Python Console内分别输入import torch和torch.cuda.is_available()进行测试 在Python Console内输入代码,在右边可以看到相应的属性 jupyter配置 安装了anaconda就已经安装了jupyter,但是此时在pytorch虚拟环境中是无法使原创 2022-05-10 16:03:08 · 655 阅读 · 0 评论 -
1.PyTorch环境的配置及安装(Configuration and Installation of PyTorch)
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】 安装anaconda 官网 视频中所用python版本为3.6,因此需要到anaconda历史版本中找到适配相应版本的安装包 对应地址 文件地址可以自己设置,其余选项默认即可 显卡准备 查看显卡驱动是否正确安装了 打开任务管理器 如果能显示显卡型号则说明显卡驱动已经正确安装了 如果没有则需要去官网下载 管理环境 conda create -n pytorch python=3.6创建专门的虚拟环境 conda acti原创 2022-05-10 15:14:15 · 503 阅读 · 0 评论