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 · 1475 阅读 · 0 评论 -
4.神经网络
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P16~24nn.Module神经网络的基本骨架,所有神经网络模块的基类。官方文档例子import torchfrom torch import nnclass Tudui(nn.Module): def __init__(self): # 这里是有操作的,code -> generate -> override methods -> --init__(self.Module) 可以快速重写这部原创 2022-05-11 16:52:07 · 513 阅读 · 0 评论 -
3.PyTorch加载数据
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P6~15Dataset类Dataset类一个表示数据集的抽象类torch.utils.data.Dataset通过继承Dataset类可以实现数据集的加载,定义三个函数,分别是:初始化 __init__()、获得每一个数据 __getitem__()、数据长度 __len__()。这里面的过程,要很清楚:__init__()用于初始化数据集的配置,例如数据集的路径、文件名称;__getitem__()用于获取给定键对应的数据原创 2022-05-10 23:11:35 · 833 阅读 · 0 评论 -
2. Python编辑器的选择、安装及配置(PyCharm、Jupyter安装)
参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】P2~5PyCharm安装pyCharm官网这里安装的是专业版其他一路默认安装即可PyCharm配置创建好项目后在Python Console内分别输入import torch和torch.cuda.is_available()进行测试在Python Console内输入代码,在右边可以看到相应的属性jupyter配置安装了anaconda就已经安装了jupyter,但是此时在pytorch虚拟环境中是无法使原创 2022-05-10 16:03:08 · 675 阅读 · 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 · 516 阅读 · 0 评论