pytorch
瓦力冫
喜欢看点书,跑跑步,热爱游戏编程
展开
-
pytorch 回归
import torchfrom torch.autograd import Variableimport torch.nn.functional as Fimport matplotlib.pyplot as pltx = torch.unsqueeze(torch.linspace(-1, 1, 100), dim = 1)y = x.pow(2) + 0.2*torch.r...原创 2018-05-19 15:07:39 · 280 阅读 · 0 评论 -
pytorch 把MNIST数据集转换成图片和txt
1.下载Mnist 数据集import os# third-party libraryimport torchimport torch.nn as nnfrom torch.autograd import Variableimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as plt...原创 2018-05-20 09:24:01 · 2647 阅读 · 6 评论 -
pytorch 把MNIST数据集转换成图片和txt
# library# standard libraryimport os# third-party libraryimport torchimport torch.nn as nnfrom torch.autograd import Variablefrom torch.utils.data import Dataset, DataLoaderimport torchvision...原创 2018-05-20 13:29:30 · 1651 阅读 · 1 评论 -
pytorch 保存和提取网络状态
# library# standard libraryimport os# third-party libraryimport torchimport torch.nn as nnfrom torch.autograd import Variablefrom torch.utils.data import Dataset, DataLoaderimport torchvision...原创 2018-05-20 15:14:22 · 2137 阅读 · 0 评论