包含标签:PyTorch 的文章
-
基于Pytorch的神经网络之Optimizer
目录 1.引言 2.各种优化器 2.1 SGD 2.2 Momentum 2.3 RMSprop 2.4 Adam 3.效果对比 1.引言 我们之前提到进行反向传播时我们需要用到优化器(Optimizer)来优…… -
Pytorch教程[03]transforms
一.torchvision:计算机视觉工具包 torchvision.transforms : 常用的图像预处理方法 torchvision.datasets : 常用数据集的dataset实现,MNIST,CIFAR-10,I…… -
Pytorch Tensor 维度操作的形象理解 Tensor.unsqueeze() Tensor.squeeze()
我们认为数组、矩阵、张量都是有形状的,假如有一个形状是 (2×2×3) 的张量 a,从左到右称为第0维、第1维、第2维…我可以使用 a[1] 取出第0维视角下的第1组数据…… -
-
Python可视化-当Matplotlib遇上matplotx
首发地址:当Matplotlib遇上matplotx 本次介绍Matplotlib的优质扩展-matplotx; matplotx在一些方面拯救Matplotlib的不完美; 同类型文章: Matplotlib-rcPar…… -
如何在STM32上部署卷积神经网络(纯C语言搭建)
0、前言 本文是什么 假如你已经使用PyTorch或者TensorFlow训练了一个卷积神经网络,得到了各层参数,却希望用C语言把这个部署到STM32等单片机上,那么就可…… -
-
-
图像中的注意力机制详解(SEBlock | ECABlock | CBAM)
图像中的注意力机制详解 注意力机制目前主要有通道注意力机制和空间注意力机制两种 一、 前言 我们知道,输入一张图片,神经网络会提取图像特征,每一…… -
ConViT
""" ConViT Model @article{d2021convit, title={ConViT: Improving Vision Transformers with Soft Convolutional Inductive Biases}, author={d'Ascol……