![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
zhuyijun09
这个作者很懒,什么都没留下…
展开
-
用pytorch写一个简单的线性回归模型
import torch import torch.nn as nn import numpy as np import matplotlib.pyplot as plt from torch.autograd import Variable 定义超参数 input_size = 1 output_size = 1 num_epochs = 1000 # 训练次数 learning_rate =...原创 2019-06-16 14:52:39 · 399 阅读 · 0 评论 -
如何使用pytorch建立逻辑回归模型
-- coding:utf-8 -- import torch import torch.nn as nn import torchvision.datasets as dsets import torchvision.transforms as transforms from torch.autograd import Variable 模型参数 单个图像大小28*28=784 input_si...原创 2019-06-23 14:20:50 · 551 阅读 · 0 评论