Pytorch demo(三)之蚂蚁和蜜蜂

模型

resnet18

在这里插入图片描述

18 = 17个卷积层 + 1个全连接层

流程

在这里插入图片描述

加载数据集并训练

此demo是区分蚂蚁和蜜蜂。
在这里插入图片描述

分类测试

【1】蚂蚁

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

【2】蜜蜂

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
PyTorch是一个开源的深度学习框架,它提供了丰富的工具和库,用于构建深度学习模型。在PyTorch中,我们可以通过构建神经网络层、优化器和损失函数来创建深度学习模型。下面就是一个PyTorch深度学习的demo。 首先,我们导入PyTorch库,然后定义一个简单的神经网络模型。比如说,我们可以创建一个包含两个全连接层的神经网络,其中第一个全连接层的输入维度是特征的维度,输出维度是隐藏层的大小,然后采用激活函数例如ReLU;第二个全连接层的输入维度是隐藏层的大小,输出维度是我们所需要的输出的大小。 接下来,我们定义损失函数和优化器。损失函数用于计算模型的预测输出和真实标签之间的差异,而优化器用于更新神经网络中的权重和偏置,以减小损失函数。在这个例子中,我们可以选择使用交叉熵损失函数和随机梯度下降优化器。 最后,我们加载数据集,将数据集分为训练集和测试集。然后我们可以迭代训练模型,在每一个epoch中,我们将训练数据输入到模型中,计算损失值,反向传播更新模型参数,然后在测试集上评估模型的性能。 通过这个简单的demo,我们可以了解如何使用PyTorch构建一个深度学习模型,并了解如何定义神经网络、损失函数和优化器,以及如何使用数据集训练和评估模型。希望这个demo可以帮助你入门PyTorch深度学习框架。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值