- 博客(11)
- 收藏
- 关注
RSS订阅原创 利用GPU进行训练
一般来说，我们可以对：网络的模型，损失函数，数据集，利用GPU进行运算也就是调用cuda.第一种方式：my_model = Model()if torch.cuda.is_available(): my_model = my_model.cuda()loss_cross = nn.CrossEntropyLoss()if torch.cuda.is_available(): loss_cross = loss_cross.cuda()imgs, targets
2021-11-30 11:41:29
1916
原创 完整的模型训练套路
保存模型方式1：vgg_16 = torchvision.models.vgg16(pretrained=False)torch.save(vgg_16, "vgg16_method1.pth")利用torch.save方法进行保存，参数2是保存的路径及其名称，该方法保存模型，保存了模型的结构及其权重参数。加载模型：...
2021-11-30 10:54:07
1121
原创 简单的搭建一个完整的网络
class Model(nn.Module): def __init__(self): super(Model, self).__init__() self.model1 = nn.Sequential( nn.Conv2d(3, 32, kernel_size=5, stride=1, padding=2), nn.MaxPool2d(kernel_size=2), nn.Conv2d(32,.
2021-11-29 11:39:35
900
原创 损失函数与反向传播
import torchinputs = torch.tensor([1, 2, 3], dtype=float)targets = torch.tensor([4, 5, 6], dtype=float)loss = torch.nn.L1Loss(reduction='sum')output = loss(inputs, targets)print(output)L1损失函数的用法，L1损失函数可以求绝对值之和或者是绝对值平均，默认是求绝对值平均。loss_mse = t.
2021-11-29 11:04:00
1649
原创 神经网络组件
池化层；class Model(nn.Module): def __init__(self): super(Model, self).__init__() self.Maxpool1 = nn.MaxPool2d(kernel_size=3, ceil_mode=True) def forward(self, input): output = self.Maxpool1(input) return o...
2021-11-27 14:48:36
340
原创 神经网络的基本框架
import torchimport torch.nn as nnclass Model(nn.Module): def __init__(self) -> None: super().__init__() def forward(self, input): output = input + 1 return outputmodel = Model()x = torch.tensor(1.0)output = mod.
2021-11-26 16:06:44
1289
原创 torchvision中的数据集使用和DATALODAER使用
import torchvisionfrom torch.utils.tensorboard import SummaryWriterdata_transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])train_set = torchvision.datasets.CIFAR10(root="./datasets", transform=data_transform, train=True, do.
2021-11-26 11:37:06
373
原创 常见的transforms
from torch.utils.tensorboard import SummaryWriterfrom PIL import Imagefrom torchvision import transforms# 将图片转换成tensorimg = Image.open("hymenoptera_data/train/ants/0013035.jpg")trans2tensor = transforms.ToTensor()img2tensor = trans2tensor(img)wri.
2021-11-25 16:06:15
1642
原创 Pytorch学习笔记
dir()函数： 查看某个package里面有哪些函数。help()函数： 查看某个函数的用法。__getitem__():按索引返回属性值class datasets: def __init__(self, name, age): self.name = name self.age = age def __getitem__(self, key): return self.namea ...
2021-11-24 19:08:13
426
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人