![](https://img-blog.csdnimg.cn/direct/51c92ed32fc242fea0c1e9917cebebdf.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch学习
记录Pytorch学习过程中的知识点等
张飞飞飞飞飞
Hello World
展开
-
Pytorch学习-引言
PyTorch是一个开源的机器学习库,广泛用于计算机视觉和自然语言处理等应用。它由Facebook的人工智能研究团队开发,并得到了许多研究机构和企业的支持。PyTorch以其易用性、灵活性和高效性而受到研究人员和开发者的青睐。动态计算图(Dynamic Computation Graph):也称为自动微分系统,它允许用户在运行时修改图形,并且可以按需进行更改。这使得模型的原型设计和调试更加直观和灵活。强大的GPU加速。原创 2024-05-14 12:52:16 · 833 阅读 · 0 评论 -
Pytorch学习-利用Dataset类定义自己的数据集
定义自己的数据集类需要继承torch.utils.data中的Dataset类。主要实现两个方法,即__len__和__getitem__以具体的例子进行演示。原创 2024-05-14 12:28:19 · 172 阅读 · 0 评论 -
Pytorch学习-模型的保存与加载
陷阱:在用此种保存方法加载模型时,需要把该模型包含的类导入到py文件中,比较麻烦,否则无法加载,或者用from ...import...的方式导入。原创 2024-05-14 10:21:24 · 212 阅读 · 0 评论 -
Pytorch学习-调整torchvision.models中模型输出类别数
假设你的类别只有10个,而torchvision.models中Vgg16的输出类别为1000,这时应该如何调整呢?方法二,再模型的最后增加全连接层,改变输出类别。原创 2024-05-14 11:09:43 · 199 阅读 · 1 评论 -
Pytorch学习-项目实战-CIFAR10图像分类
由于CIFAR数据集具有10个类别,因此最后的全连接层要为10。原创 2024-05-14 10:35:52 · 188 阅读 · 0 评论