pytorch
wangzai6378
喜欢游戏;但却没有做游戏
展开
-
pytorch 一元线形回归
今天做的是一个用pytorch训练出来的一元线形回归模型;如图,我们先找一些样本点,用 matplotlib.pyplot显示如上,我们把它加载到内存里面,然后通过pytorch的模组简单自定义一下前向函数就可以了,完整代码如下 #!/usr/bin/python # -*- coding: UTF-8 -*- import torch from torch.autogr...原创 2019-04-14 16:48:30 · 292 阅读 · 0 评论 -
pytorch minist全网络手写数字识别
#!/usr/bin/python # -*- coding: UTF-8 -*- import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader from torch.autograd import Variable import torch.nn.functi...原创 2019-05-11 08:59:19 · 597 阅读 · 0 评论 -
Pytorch实现简单的CNN网络
#!/usr/bin/python # -*- coding: UTF-8 -*- import torch from torchvision import datasets, transforms from torch.utils.data import DataLoader from torch.autograd import Variable import torch.nn.functi...原创 2019-06-02 11:18:24 · 1795 阅读 · 1 评论