码迷,mamicode.com
首页 >  
搜索关键字:pytorch    ( 1103个结果
pytorch中的torch.autograd.backward()和torch.autograd.grad()
参考 一文读懂torch.autograd.backward()作用及原理 torch.autograd学习系列之torch.autograd.grad()函数学习 主要是对这两个函数中的参数grad_tensors或grad_outputs不理解。其实就是雅克比矢量积的系数,有了这个系数就能将矢量 ...
分类:其他好文   时间:2021-01-20 11:39:16    阅读次数:0
【深度强化学习】9. Policy Gradient实现中核心部分torch.distributions
【导语】:在深度强化学习第四篇中,讲了Policy Gradient的理论。通过最终推导得到的公式,本文用PyTorch简单实现以下,并且尽可能搞清楚torch.distribution的使用方法。代码参考了LeeDeepRl-Notes中的实现。 1. 复习 \[ \theta \leftarro ...
分类:其他好文   时间:2021-01-16 12:16:36    阅读次数:0
pytorch模型结构可视化方法
最近在学习一些检测方面的网络,使用的是pytorch。模型结构可视化是学习网络的有用的部分,pytorch没有原生支持这个功能,需要找一些其他方式,下面总结几种方法(推荐用4)。 1. torch .pt -> netron netron是一个专门可视化模型的工具,支持很多格式,很自然的就是用它直接 ...
分类:其他好文   时间:2021-01-08 11:33:12    阅读次数:0
[pytorch] API总结、速查
torch.numel(input) → int Returns the total number of elements in the input tensor. Document torch.from_numpy(ndarray) → Tensor Creates a Tensor from a ...
分类:Windows程序   时间:2021-01-05 11:40:06    阅读次数:0
win10+cuda10.2安装pytorch1.7.1(使用清华镜像源)
1、查看自己的设备对应的cuda版本 (1)打开NVIDIA控制面板 (2)点击“帮助”,如下图: (3)点击“系统信息” >“组件”可以看到下图,CUDA版本是10.2: 2、在官网(https://pytorch.org/)上查看对应的版本 从下图可以看到红线标注的是需要运行的命令。但是,如果直 ...
分类:Windows程序   时间:2021-01-04 11:01:56    阅读次数:0
使用anaconda安装pytorch
更换清华镜像 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.e ...
分类:其他好文   时间:2020-12-25 13:03:52    阅读次数:0
pytorch部署torchscript篇
title: pytorch部署torchscript篇 date: 2020 12 16 22:23:29 tags: pytorch categories: pytorch comments: true 引言 本文旨在介绍如何在Windows平台使用pytorch的c++ api部署pytorc ...
分类:其他好文   时间:2020-12-25 12:00:26    阅读次数:0
pytorch训练AlexNet
一。AlexNet网络结构和参数 二。训练部分 model.py 1 import torch.nn as nn 2 import torch 3 4 5 class AlexNet(nn.Module): 6 def __init__(self, num_classes=1000, init_we ...
分类:Web程序   时间:2020-12-25 11:37:31    阅读次数:0
Google Cloud TPUs支持Pytorch框架啦!
文|Sherry在2019年PyTorch开发者大会上,Facebook,Google和SalesforceResearch联合宣布启动PyTorch-TPU项目。项目的目标是在保持PyTorch的灵活性的同时让社区尽可能容易地利用云TPU提供的高性能计算。团队创建了PyTorch/XLA这个repo,它可以让使PyTorch连接到云TPU并使用TPU内核,同时Colab也支持在云TPU上使用Py
分类:其他好文   时间:2020-12-24 12:11:57    阅读次数:0
Anaconda安装pytorch
更换清华源的时候,这三个都需要加上 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tun ...
分类:其他好文   时间:2020-12-24 12:05:29    阅读次数:0
1103条   上一页 1 ... 5 6 7 8 9 ... 111 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!