ICode9

精准搜索请尝试: 精确搜索
  • 自定义打印类信息:def __repr__(self)2022-03-28 15:32:05

    Example: import torch import torch.nn.functional as F from torch.nn.modules.module import Module from torch.nn.parameter import Parameter class GraphConvolution(Module): def __init__(self, in_features, out_features, dropout=0., act=F.relu):

  • 移动云使用经验(三)安装marl所需包 (pytorch, gym, mpe)2022-03-27 21:01:09

    1 安装pytorch 首先按照(二)用conda新建虚拟环境,建议用python3.8,首先激活虚拟环境并参考博文添加清华源,然后按照官网运行如下指令: conda activate #env_nameconda install pytorch=1.10.0 cudatoolkit=10.2 我只安装了torch。(博客写于20220228,未来指令可能会改变)  然后安装torch_

  • torch.randn(1,3,8,9)2022-03-27 16:32:25

    input = torch.randn(1, 3, 8, 9) print(input) 输出:1这个张量,其中是3个通道,8行,9列的数据(从外面往里面数) tensor([[[[-1.3645, 1.3841, 0.9907, 0.3150, -0.2379, -0.3170, -0.0550, 1.1550, -0.3213], [-0.1501, 1.7308, 1.8265, -1.2117, -0.4732, 0.93

  • pytorch基本语法2022-03-26 17:00:08

    pytorch安装 bing官网 找对应版本下载 记住一点:-c pytorch不要加上,否则下载速度巨慢 pytorch的基本语法 tensor张量 基本概念 标量:0维张量 矢量:一维张量 矩阵:二维张量 矩阵数组:三维张量 tensor()是一个函数 代码演示: tensor = torch.arange(2,14,2):不用手打,创建一维张量,取不到14

  • Pytorch之torch.meshgrid()2022-03-21 17:04:38

    说明:   torch.meshgrid()的功能是生成网格,可以用于生成坐标。 函数输入:   输入两个数据类型相同的一维tensor 函数输出: 输出两个tensor(tensor行数为第一个输入张量的元素个数,列数为第二个输入张量的元素个数) 注意:   1)当两个输入tensor数据类型不同或维度不是一维时会报错

  • Pytorch实现FCN中对损失函数的理解2022-03-21 12:02:55

    代码:https://github.com/bat67/pytorch-FCN-easiest-demo FCN网络的输出 输入网络的图片大小为H×W,通道数为3(RGB图像)。 而经过卷积和上采样过程,输出图片大小不变,仍为H×W,通道数为分割的类别数C。 在Pytorch中输出的shape为torch.Size([C, H, W]) FCN网络的损失函数 criter

  • 深度学习图像处理目标检测图像分割计算机视觉 11--医疗影像分割2022-03-20 20:59:46

    深度学习图像处理目标检测图像分割计算机视觉 11--医疗影像分割 摘要一、医疗影像分割二、U-Net三、Brain Tumor Classification based on MR Images using GAN as a Pre-Trained Model基于 MR 图像的脑肿瘤分类使用 GAN 作为预训练模型3.1、摘要3.2、介绍3.3、模型、结构3

  • 将强化学习应用到量化投资中实战篇(神经网络模块开发)2022-03-20 18:02:31

    将强化学习应用到量化投资中实战篇(神经网络模块开发) 文章目录 将强化学习应用到量化投资中实战篇(神经网络模块开发)1.前言2 父类Network类3. DNN4.LSTM5.CNN 1.前言 在本节内容中,将详细介绍神经网络模块中包含的几个神经网络类的属性和功能,并详细讲解基于 Python和Pytor

  • 神经网络参数初始化参数固定2022-03-19 19:35:18

    一般来说,神经网络模型实验重复多次,但最终结果仍然有较大波动时,应该从三个方面结果:种子、初始点、优化器。 其中种子的固定尤为关键,应在代码中加入下列种子固定代码 def setup_seed(seed): random.seed(seed) np.random.seed(seed) torch.manual_seed(seed) tor

  • FairMOT复现2022-03-19 13:32:29

    FairMOT复现 根据https://www.bilibili.com/video/BV1RD4y1S7ws这位UP主的教程将代码跑通,在此记录一下。 环境 显卡:3060 cuda版本: 11.2 pytorch版本:1.9.0 python版本: 注意请使用python3.7,之前用1.8报过一些numpy的版本错误问题。 编译器:Pycharm 3060cuda配置可以参考这

  • RuntimeError: CUDA error: no kernel image is available for execution on the device2022-03-19 11:59:01

    RuntimeError: CUDA error: no kernel image is available for execution on the device 这个问题是cuda版本和torch版本不对应,如下图,我的服务器cuda版本是11.2: 我复现的代码要求的torch版本是1.6.0,但是查找pytorch官网发现,torch1.6.0对应的cuda最高只到10.2,如图:   所以,只能

  • pytorch下载自带数据集并transform2022-03-19 11:33:05

    import os.path import numpy as np import torch import cv2 from PIL import Image from torch.utils.data import Dataset import re from functools import reduce from torch.utils.tensorboard import SummaryWriter as Writer from torchvision import transforms imp

  • Pytorch中with torch.no_grad()或@torch.no_grad() 用法2022-03-19 10:31:43

    参考Pytorch中with torch.no_grad()或@torch.no_grad() 用法 - 云+社区 - 腾讯云 requires_grad=True 要求计算梯度 requires_grad=False 不要求计算梯度 with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播 model.eval()

  • 《深度学习框架PyTorch入门与实践》——Tensor基本操作(2)2022-03-18 17:10:28

    一、Tensor操作 ************************************************************************************************************** (1)归并操作      函数功能mean/sum/median/mju均值/和/中位数/众数nom/dist范数/距离std/var标准差/方差cumsum/cumprod累加/累乘 import

  • torch保存加载模型2022-03-09 22:00:49

    目录三个核心函数状态字典定义只保存/加载模型参数(推荐做法)保存/加载整个模型断点训练checkpoint使用同一个文件中保存多个模型用一个模型的参数来初始化另一个不同模型不同设备保存/加载模型 三个核心函数 torch.save() torch.load() torch.nn.Module.load_state_dict() 状态字

  • with torch.no_grad()2022-03-09 11:03:06

    在pytorch中,tensor有一个requires_grad参数,如果设置为True,则反向传播时,该tensor就会自动求导。tensor的requires_grad的属性默认为False,若一个节点(叶子变量:自己创建的tensor)requires_grad被设置为True,那么所有依赖它的节点requires_grad都为True(即使其他相依赖的tensor的requires

  • Pytorch 损失函数总结2022-03-09 09:32:58

    1 nn.L1Loss   torch.nn.L1Loss(reduction='mean')   就是 MAE(mean absolute error),计算公式为     $\ell(x, y)=L=\left\{l_{1}, \ldots, l_{N}\right\}^{\top}, \quad l_{n}=\left|x_{n}-y_{n}\right|$     $\ell(x, y)=\left\{\begin{array}{ll}\opera

  • pytorch 保存训练好的模型2022-03-08 17:00:46

    1 保存和加载整个模型  torch.save(model_object, 'model.pth') model = torch.load('model.pth') 2 仅保存和加载模型参数 torch.save(model_obj.state_dict(), 'params.pth') model_obj.load_state_dict(torch.load('params.pth')) 3 选择保存网络中

  • torch.optim optimizer函数2022-03-08 12:35:24

    class torch.optim.SGD(params, lr=, momentum=0, dampening=0, weight_decay=0, nesterov=False)[source] 实现随机梯度下降算法(momentum可选)。 Nesterov动量基于On the importance of initialization and momentum in deep learning中的公式.   参数: params (iterable) – 待

  • torch.randn和torch.rand函数2022-03-08 12:33:52

    1.均匀分布torch.rand(*sizes, out=None) → Tensor 返回一个张量,包含了从区间[0, 1)的均匀分布中抽取的一组随机数。张量的形状由参数sizes定义。 参数:  sizes (int...) - 整数序列,定义了输出张量的形状; out (Tensor, optinal) - 结果张量 例子:torch.rand(2, 3) 0.0836 0.615

  • 手动实现前馈神经网络解决 二分类 任务2022-03-06 12:04:09

    1 导入实验需要的包 import numpy as np import torch from torch import nn from sklearn.model_selection import train_test_split import matplotlib.pyplot as plt from torch.utils.data import TensorDataset,DataLoader 2 创建数据 num_example,num_input = 10000,200

  • Pytorch nn.BCEWithLogitsLoss() 的简单理解与用法2022-03-06 10:32:21

    这个东西,本质上和nn.BCELoss()没有区别,只是在BCELoss上加了个logits函数(也就是sigmoid函数),例子如下: import torch import torch.nn as nn label = torch.Tensor([1, 1, 0]) pred = torch.Tensor([3, 2, 1]) pred_sig = torch.sigmoid(pred) loss = nn.BCELoss() print(loss(pr

  • 深度学习准确率(acc)计算方法2022-03-06 09:00:22

    本文简单介绍模型训练时候,使用准确率求解过程,不涉及精确率和召回率计算, 本文给出简要计算方法与代码。   计算方法: 使用top1计算为例(以下以2个batch,3个num_classes举列): 网络预测结果形式:pred=[b,num_classes] ,如pred=[[0.6,0.8,0.9],[0.7,0.4,0.3]] 真实标签形式:label=[b],如bat

  • 手动实现前馈神经网络解决 多分类 任务2022-03-06 01:32:50

    1 导入实验需要的包 import torch import numpy as np import random from IPython import display import matplotlib.pyplot as plt from torch.utils.data import DataLoader,TensorDataset from torchvision import transforms,datasets from torch import nn 2 加载数据集

  • PyTorch + OpenVINO 开发实战系列教程 第一篇2022-03-03 14:05:10

    第1章 Pytorch介绍与基础知识 大家好,本章是主要介绍一下深度学习框架Pytorch的的历史与发展,主要模块构成与基础操作代码演示。重点介绍Pytorch的各个组件、编程方式、环境搭建、基础操作代码演示。本章对有Pytorch开发经验的读者来说可以直接跳过;对初次接触Pytorch的读者来说,通

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有