pytorch学习
马飞飞
乐于分享,痴迷技术
展开
-
pytorch-神经网络拟合曲线
代码已经调通,跑出来的效果如下:# coding=gbkimport torchimport matplotlib.pyplot as pltfrom torch.autograd import Variableimport torch.nn.functional as F''' Pytorch是一个拥有强力GPU加速的张量和动态构建网络的库,其主要构建是张量,所以可...原创 2018-11-26 00:02:12 · 9427 阅读 · 5 评论 -
pytorch-RNN进行回归曲线预测
任务通过输入的sin曲线与预测出对应的cos曲线#初始加载包 和定义参数import torchfrom torch import nnimport numpy as npimport matplotlib.pyplot as plttorch.manual_seed(1) #为了可复现#超参数设定TIME_SETP=10INPUT_SIZE=1LR=0.02D...原创 2018-11-26 00:06:46 · 6145 阅读 · 1 评论