pcmvay的博客

话不多说我们直接先把完整的代码贴出来import torchimport torchvisionfrom torch.utils.tensorboard import SummaryWriter# from model import *# 准备数据集from torch import nnfrom torch.utils.data import DataLoader# 定义训练的设备device = torch.device("cuda")train_data = tor...

Pytorch官网上给我们提供了很多已训练好或未训练的网络模型，我们可以直接来进行使用 import torchvision.models as modelsresnet18 = models.resnet18()alexnet = models.alexnet()vgg16 = models.vgg16()squeezenet = models.squeezenet1_0()densenet = models.densenet161()inception = models....

接触机器学习和深度学习已经有一段时间了，一直想做个记录，方便自己以后的查阅。 一开始我搭建神经网络时所使用的框架是Tensorflow，虽然功能强大但是不同版本代码的兼容性有些差强人意。 以下的内容所创建的环境是Anaconda中的虚拟环境，采用的python版本是3.8，cuda和cudnn都是对应的版本。 搭建和训练神经网络分为以下几个步骤：1.数据集的准备 2.神经网络的搭建 3.反向传播调整参数降低loss值 只是一个记录。 一. 数据集的准备 搭建完一个神经...