PyTorch
文章平均质量分 53
Hong_Youth
热爱生活、认真工作、成长进步!
人工智能、计算机视觉、编程语言等方向技术分享。
展开
-
一文看懂网络中间层特征矩阵和卷积层参数
在使用卷机神经网络训练模型时,很多时候需要查看卷积层的特征矩阵是什么样子的,看它提取的是什么特征,有多少卷积层参数以及将它可视化出来,本期本文就给大家讲解一下怎么使用Pytorch和TensorFlow查看中间层的特征矩阵和卷积层参数。1、Pytorch查看特征图和卷积参数我们以AlexNet这个简单的卷积神经网络为例来查看它的中间层特征矩阵和卷积层参数。首先建立我们的model.py,在正向传播函数中保存需要显示的卷积层输出的特征矩阵在一个列表中。import torch.nn as n原创 2021-05-02 13:12:12 · 1352 阅读 · 7 评论 -
Python中PyTorch环境配置
python里面配置PyTorch环境有两种方式,一种使用Anaconda里面的conda命令配置;一种是使用pip命令配置。进入PyTorch官网,依次选择你电脑的配置(我这里已经下载了python3.7)1.pip方式需要下载GPU版本的可以选择对应cuda版本的命令;后面复制命令在控制台运行即可。2.conda方式以然可以选择gpu版本,复制命令到控制台运行(前提是已经安装好了Anaconda,并配置好了环境变量)测试是否安装好了PyTorch:...原创 2020-09-11 21:44:51 · 1437 阅读 · 0 评论 -
PyTorch的说明和介绍
简介PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序。它主要由Facebookd的人工智能小组开发,不仅能够 实现强大的GPU加速,同时还支持动态神经网络,这一点是现在很多主流框架如TensorFlow都不支持的。 PyTorch提供了两个高级功能:1. 具有强大的GPU加速的张量计算(如Numpy)2. 包含自动求导系统的深度神经网络除了Facebook之外,Twitter、GMU和Salesforce等机构都采用了PyTorch。TensorFlow和原创 2020-09-11 21:21:22 · 1256 阅读 · 0 评论