摘要: 使用VGG模型进行猫狗大战 首先判断是否存在GPU设备 2. 数据处理 datasets 是 torchvision 中的一个包,可以用做加载图像数据。它可以以多线程(multi-thread)的形式从硬盘中读取数据,使用 mini-batch 的形式,在网络训练中向 GPU 输送。在使用CNN处理 阅读全文
posted @ 2021-10-24 19:44 Drew。 阅读(36) 评论(0) 推荐(0)
摘要: 卷积神经网络学习 首先加载CIFAR10使用torchvision。 1. 加载数据 (MNIST) PyTorch里包含了 MNIST, CIFAR10 等常用数据集,调用 torchvision.datasets 即可把这些数据由远程下载到本地。 2. 创建网络 定义网络时,需要继承nn.Mod 阅读全文
posted @ 2021-10-17 20:35 Drew。 阅读(41) 评论(0) 推荐(0)
摘要: 第一次深度学习 1.PyTorch基本操作练习 1.定义数据 创建一些张量 2.定义操作 凡是用Tensor进行各种运算的,都是Function 最终,还是需要用Tensor来进行计算的,计算无非是 基本运算,加减乘除,求幂求余 布尔运算,大于小于,最大最小 线性运算,矩阵乘法,求模,求行列式 基本 阅读全文
posted @ 2021-10-09 20:30 Drew。 阅读(55) 评论(0) 推荐(0)