torch 的 forward 和 backward
2018-01-01 11:31
253 查看
Criterions有其forward和backward函数
https://github.com/torch/nn/blob/master/doc/criterion.md
Module也有其forward和backward函数
https://github.com/torch/nn/blob/master/doc/module.md
Module的forward函数最简单,就是输入input得到output
Module的backward看下这个线性回归的例子
https://github.com/torch/nn/blob/master/doc/criterion.md
Module也有其forward和backward函数
https://github.com/torch/nn/blob/master/doc/module.md
Module的forward函数最简单,就是输入input得到output
Module的backward看下这个线性回归的例子
require 'torch' require 'nn' require 'gnuplot' month = torch.range(1,10) price = torch.Tensor{28993,29110,29436,30791,33384,36762,39900,39972,40230,40146} model = nn.Linear(1, 1) criterion = nn.MSECriterion() month_train = month:reshape(10,1) price_train = price:reshape(10,1) for i=1,1000 do price_predict = model:forward(month_train) -- 输入 -> 输出 err = criterion:forward(price_predict, price_train) -- 输出,正确 -> loss值 print(i, err) model:zeroGradParameters() gradient = criterion:backward(price_predict, price_train) -- 输出,正确 -> 梯度 model:backward(month_train, gradient) -- 输入,梯度 model:updateParameters(0.01) end month_predict = torch.range(1,12) local price_predict = model:forward(month_predict:reshape(12,1)) print(price_predict) gnuplot.pngfigure('plot.png') gnuplot.plot({month, price}, {month_predict, price_predict}) gnuplot.plotflush()
相关文章推荐
- Brief overview of backward and forward
- DFS遍历中forward、backward以及cross边的界定
- Highly Efficient Forward and Backward Propagation of Convolutional Neural Networks for Pixelwise Cla
- 解剖caffe:Forward 和 Backward
- Caffe学习:Forward and Backward
- Andrew Ng 深度学习课程Deeplearning.ai 编程作业——forward and backward propagation(1-4.1)
- Caffe添加层GPU实现.cu文件,提示无法解析Forward_gpu(...)、Backward_gpu(...)问题解决
- Pytorch入门学习(八)-----自定义层的实现(甚至不可导operation的backward写法)
- Codeforces Gym 100889 B Backward and Forward
- caffe study(2) 关于forward和backward - backward
- pytorch学习笔记(七):pytorch hook 和 关于pytorch backward过程的理解
- torch.autograd.backward中的参数问题
- Caffe学习:Forward and Backward
- PEGA之Declare-Expression 与Forward-Chaining/Backward-Chaining ---- RichardLu
- Sql Server 聚集索引扫描 Scan Direction的两种方式------FORWARD 和 BACKWARD
- PEGA之Declare-Expression 与Forward-Chaining/Backward-Chaining ---- RichardLu
- pytorch学习笔记(七):pytorch hook 和 关于pytorch backward过程的理解
- Caffe入门(4)——Forward and Backward
- Feed Forward and Backward Run in Deep Convolution Neural Network 论文阅读笔记
- gstreamer play speed fastforward fastbackward