神经网络框架
npupengsir
新手程序员
展开
-
pytorch模型可视化
1. 使用dot1.1 安装graphviz和torchvizsudo apt-get install graphvizsudo pip install torchviz1.2 使用torchvizimport torchfrom torch import nnfrom torchviz import make_dot, make_dot_from_trace# Visualize gradients of simple MLP# The method below is for bu原创 2020-10-27 11:28:18 · 1175 阅读 · 0 评论 -
windows安装caffe
caffe windows上有教程,但是有几点没有详细说明。1. 安装vc编译器安装vc编译器后,在cmakelist.txt中设置使用vc编译器而不是gcc:set(CMAKE_C_COMPILER "D:/Microsoft Visual Studio 12.0/VC/bin/cl.exe")set(CMAKE_CXX_COMPILER "D:/Microsoft Visual Studio 12.0/VC/bin/cl.exe")2. 安装boost在boost官网上下载exe安装文件。原创 2020-10-17 17:09:14 · 295 阅读 · 0 评论 -
PyTorch入门
详见官方文档: pytorch1. PyTorch中的Variable和TensorPyTorch中的Tensor就是一个高维数组,类似于numpyarray。而Variable是对Tensor的一个封装,有3个对应的属性:v.data对应的是tensor本身,v.grad是对应的tensor的梯度以及v.grad_fn。###2. PyTorch中的主要模块torch: 顶层PyTroch包和tensor库torch.nn: 构建神经网络的子库torch.autograd: 支持可微tens原创 2020-06-10 03:14:00 · 304 阅读 · 0 评论 -
tensor flow学习笔记
1. Tensor的概念在TensorFlow中,一切数据都是张量(tensor),是vector和matrices的泛化,例如:vector是一个1D的tensor, matrix是一个2维的tensortf.constant(value, dtype=None, shape=None, name=’const’) 创建一个常量tensor, 按照给出的value赋值,value可以是一个数,也原创 2017-10-14 15:48:09 · 2269 阅读 · 0 评论 -
keras学习笔记
具体请查看官方文档keras官方文档1.基本概述neural layers(神经层), cost functions(损失函数), optimizers(优化器), initialization schemes(初始化方案), activation functions(激活函数), regularization(正则化项)在keras中都是独立的模块,可以自由组合。keras的核心...原创 2017-10-06 21:42:11 · 16659 阅读 · 2 评论 -
ubuntu安装caffe
ubuntu安装caffe目前caffe只支持python2,所以使用caffe的童鞋们请使用python2安装。官方文档caffe安装1. 安装相关依赖sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-devsudo apt-get install libhdf5-serial-dev proto原创 2017-12-17 23:11:09 · 512 阅读 · 0 评论 -
绘制Caffe prototxt中网络结构图
1. 安装相关依赖安装caffe安装graphvizsudo apt-get install graphviz # 未安装可能出现 OSError: [Errno 2] "dot" not found in path.安装pydotsudo pip install pydot2. 绘制图进入到prototxt所在目录下绘图python /p原创 2018-01-18 22:09:15 · 1055 阅读 · 0 评论