![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
文章平均质量分 61
AI柱子哥
这个作者很懒,什么都没留下…
展开
-
【AI】数据集Dataloader制作
以花朵分类的数据集来进行测试。Oxford 102 Flowers Dataset 是一个花卉集合数据集,主要用于图像分类,它分为 102 个类别共计 102 种花,其中每个类别包含 40 到 258 张图像。该数据集由牛津大学工程科学系于 2008 年发布,相关论文有《Automated flower classification over a large number of classes》。原创 2023-12-01 16:53:56 · 142 阅读 · 0 评论 -
【AI】Pytorch神经网络分类初探
这边直接引用官网教程的模型device = ("cuda"else "mps"else "cpu"device")nn.ReLU(),nn.ReLU(),将打印的结果放在下面,可以查看一下这里我们依旧使用官网教程中的直接来这里的SGD是最基础的优化器,采用的是梯度递减的方式,其收敛的会比较慢,如果希望收敛快些,可以使用Adam方式。训练函数>7f>5d>5d测试函数100>0.1f>8f。原创 2023-12-01 15:30:41 · 316 阅读 · 0 评论