pytorch
五四三两幺-发射!
这个作者很懒,什么都没留下…
展开
-
windows10上安装detectron2(0.2.1) + pytorch1.5.1 + CUDA10.1
0. 安装NVIDIA显卡驱动, 安装CUDA(因为底层需要用到c++编译deformable conv层) 注意: 由于后面用conda安装pytorch会安装conda的cudatoolkit包, 需要确保cudatoolkit和这里安装的本地CUDA版本一致(比如下面装的是CUDA10.1, 这里也要装CUDA10.1)1. 根据NVIDIA显卡驱动确定自己可以安装的CUDA版本以及pytorch版本 上图来自cuda-toolkit-release-notes 比如...原创 2020-10-19 11:47:58 · 2547 阅读 · 0 评论 -
把vgg-face.mat权重迁移到pytorch模型
最近使用pytorch时,需要用到一个预训练好的人脸识别模型提取人脸ID特征,想到很多人都在用用vgg-face,但是vgg-face没有pytorch的模型,于是写个vgg-face.mat转到pytorch模型的代码#!/usr/bin/env python2# -*- coding: utf-8 -*-"""Created on Thu May 10 10:41:40 2018...原创 2018-05-17 11:00:39 · 3227 阅读 · 7 评论 -
pytorch中如何只让指定变量向后传播梯度?
pytorch中如何只让指定变量向后传播梯度?(或者说如何让指定变量不参与后向传播?)有以下公式,假如要让L对xvar求导:(1)中,L对xvar的求导将同时计算out1部分和out2部分;(2)中,L对xvar的求导只计算out2部分,因为out1的requires_grad=False;(3)中,L对xvar的求导只计算out1部分,因为out2的requires_gr...原创 2018-05-23 11:21:49 · 5017 阅读 · 3 评论 -
InforGAN笔记
InforGAN笔记Purpose: 通过非监督学习得到可分解的特征表示.Motivation:由于G使用噪声z的时候没有加任何的限制, 所以1z是在以一种高度混合的方式被使用, 2z的任何一个维度都没有被明显的表示数据的一个特征(即我们不知道什么样的噪声可以生成数字1,什么样的噪声可以生成数字3...)Solution:作者在生成器的输入中, 除了z之外还增加了一个隐含编码c(也是随机的)...原创 2018-06-07 21:06:08 · 1291 阅读 · 0 评论 -
Pytorch使用manual_seed
seed = 2018torch.manual_seed(seed)torch.cuda.manual_seed(seed)a=torch.rand([1,5])b=torch.rand([1,5])关闭进程,重新运行上面代码,发现得到的a和b和上一次的a和b是一样的seed = 2018torch.manual_seed(seed)torch.cuda.manual_se...原创 2018-06-26 16:48:24 · 14883 阅读 · 0 评论 -
CornerNet源码中对_decode函数(最后的输出处理)的理解
文章: https://arxiv.org/pdf/1808.01244.pdf源码链接: https://github.com/princeton-vl/CornerNet.git, 基于Pytorch实现_decode函数位于: CornerNet/models/py_utils/kp_utils.py中以往的目标检测框架Faster RCNN, YOLO, SSD之类的都需要在网络中生...原创 2019-04-26 10:18:48 · 1417 阅读 · 3 评论