![](https://img-blog.csdnimg.cn/277ccdd5cadd49beb45593652b837b84.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
pytorch
文章平均质量分 54
pytorch
Enzo 想砸电脑
这个作者很懒,什么都没留下…
展开
-
【无标题】
注 : 因为函数中没有指定参数 dim,所以操作是会先将 x 展平为。,再 reshape 成原来的尺寸,得到。, 移动 1 位,变成。原创 2023-07-08 12:00:06 · 212 阅读 · 0 评论 -
torch.meshgrid()
【代码】torch.meshgrid()原创 2023-07-06 23:13:01 · 488 阅读 · 0 评论 -
【pytorch】自定义层
我们先介绍如何定义一个不含参数的自定义层。事实上,创建自定义层 与 使用 Module类 构造模型类似。下面的 CenteredLayer 类通过继承 Module类 自定义了一个将输入减掉均值后输出的层,并将层的计算定义在了forward函数里。(这个层里不含模型参数)之前的文章已经介绍了模型参数。Parameter类 其实是Tensor的子类,如果一个 Tensor 是 Parameter,那么它会自动被添加到模型的参数列表里。所以在自定义含模型参数的层时,我们需要将参数定义成Parameter。原创 2023-02-21 13:17:22 · 1082 阅读 · 0 评论 -
【pytorch】保存模型、加载模型
我们可以直接使用 save 函数 和 load函数 进行存储和读取。的可学习参数 (即权重和偏差),模块模型包含在参数中 (通过。注意,只有具有可学习参数的层(卷积层、线性层等) 才有。,其中包含关于优化器状态以及所使用的超参数的信息。是一个从参数名称隐射到参数。原创 2023-02-21 12:09:31 · 776 阅读 · 0 评论 -
【迁移学习】Transfer Learning
【迁移学习】Transfer Learning原创 2022-09-07 17:51:03 · 1471 阅读 · 0 评论 -
【torchsummary】torchsummary 查看网 及 各层输出的 feature map 尺寸 (Pytorch + AlexNet)
【代码】【Pytorch + AlexNet + torchsummary】torchsummary 查看网 及 各层输出的 feature map 尺寸。原创 2022-08-29 12:55:39 · 952 阅读 · 0 评论 -
【无标题】
在PyTorch中,可以使用torchstat这个库来查看网络模型的一些信息,包括总的参数量params、MAdd、显卡内存占用量和FLOPs等。原创 2022-08-29 08:10:40 · 86 阅读 · 0 评论 -
【pytorch】torch.device
代表将分配到的设备的对象(简单点说,就是分配到你的CPU还是GPU上,以及哪块GPU上)torch.device包含一个设备类型('cpu’或’cuda’设备类型)和可选的设备的序号。如果设备序号不存在,则为当前设备例如,torch.Tensor用设备构建’cuda’的结果等同于’cuda:X’,其中X是torch.cuda.current_device()的结果。......原创 2022-06-12 11:03:54 · 1648 阅读 · 0 评论 -
【pytorch】softmax函数
做过多分类任务的同学一定都知道softmax函数。softmax函数,又称归一化指数函数。它是二分类函数sigmoid在多分类上的推广,目的是将多分类的结果以概率的形式展现出来。下图展示了softmax的计算方法:下面为大家解释一下为什么softmax是这种形式。首先,我们知道概率有两个性质:1)预测的概率为非负数;2)各种预测结果概率之和等于1。softmax就是将在负无穷到正无穷上的预测结果按照这两步转换为概率的。1)将预测结果转化为非负数下图为y=exp(x)的图像,我们可以知道指数函数的值域取值范转载 2022-06-13 07:57:26 · 4917 阅读 · 0 评论 -
【pytorch】nn.AdaptiveMaxPool2d
官方文档地址:https://pytorch.org/docs/master/generated/torch.nn.AdaptiveMaxPool2d.html#torch.nn.AdaptiveMaxPool2dtorch.nn.AdaptiveMaxPool2doutput_size – the target output size of the image of the form H x W. Can be a tuple (H, W) or a single H for a square imag原创 2022-06-09 08:34:15 · 759 阅读 · 0 评论 -
【pytorch】2.7 卷积函数 nn.conv2d
二维卷积应该是最常用的卷积方式了,在Pytorch的nn模块中,封装了nn.Conv2d()类作为二维卷积的实现。使用方法和普通的类一样,先实例化再使用。下面是一个只有一层二维卷积的神经网络,作为nn.Conv2d()方法的使用简介:它的形参由Pytorch手册可以查得,前三个参数是必须手动提供的,后面的有默认值。接下来将一一介绍:这个很好理解,就是输入的四维张量[N, C, H, W]中的C了,即输入张量的channels数。这个形参是确定权重等可学习参数的shape所必需的。也很好理解,即期望的四维原创 2022-06-05 22:48:01 · 400 阅读 · 0 评论 -
【pytorch】2.4 kaggle数据集 数据预处理、重写Dataset、DataLoader读取数据
1、从kaggle上下载相关数据:地址整个数据集包括3个部分:训练集文件夹、label 的csv文件、测试集文件夹数据下载下来后,读取相关信息看一下import pandas as pddf = pd.read_csv('/Users/manmi/Desktop/dog_breed/labels.csv')print(df.info())print(df.head())训练集一共10221条数据,id对应着图片的名称,但是不带后缀.jpg,breed对应犬种名称。.........原创 2022-05-29 21:30:34 · 1952 阅读 · 0 评论 -
【pytorch】2.3 神经网络核心组件与结构(举例: MNIST 手写数字识别)
MNIST 手写数字识别原创 2022-05-08 20:24:10 · 676 阅读 · 0 评论 -
【pytorch】2.1 使用Numpy实现 机器学习(梯度下降法)
import numpy as npfrom matplotlib import pyplot as pltnp.random.seed(100)x = np.linspace(-1, 1, 100).reshape(100, 1)y = 3 * np.power(x, 2) + 2 + 0.2 * np.random.randn(x.size).reshape(100, 1)# plt.scatter(list(x), list(y))# plt.show()w1 = np.rando原创 2022-05-08 18:39:20 · 593 阅读 · 0 评论 -
【pytorch】1.7 pytorch与numpy,tensor与array的转换
pytorch与numpy,tensor与array的转换原创 2022-07-30 23:22:06 · 1598 阅读 · 0 评论 -
【numpy】np.random
生成[low,high)之间的随机浮点数,随机的概率是均匀的。原创 2022-08-24 13:55:59 · 358 阅读 · 0 评论