记录pytorch代码运行中的报错

【1】RuntimeError: size mismatch, m1: [1 x 100], m2: [1 x 1] at c:\programdata\miniconda3\conda-bld\pytorch-cpu_1532498166916\work\aten\src\th\generic/THTensorMath.cpp:2070

x = np.linspace(-1, 1, 100)
    # 在指定的间隔内返回均匀间隔的数字
y = 2 * x + np.random.randn(*x.shape) * 0.3 # y=2x,但是加入了噪声

#def change_to_torch(a): #转成torch
x_train_x = torch.from_numpy(x)
y_train_y = torch.from_numpy(y)

x_train = x_train_x.float() #将类型固定为float 类型
y_train = y_train_y.float()

class LinearRegression(nn.Module):
    def __init__(self):
        super(LinearRegression,self).__init__()
        self.linear = nn.Linear(1,1) #输入输出纬度均为1
    
    def forward(self,x):
        out = self.linear((x))
        return out

#初始化
model = LinearRegression()  #创建模型,未实例化  
criterion = nn.MSELoss()  #定义损失函数为均方误差
optimizer = optim.SGD(model.parameters(),lr=1e-2) # 定义优化算法为梯度下降法

#训练   
num_epochs = 1000 #训练次数
for epoch in range(num_epochs):  #将tensor类型转成 variable类型
    inputs = Var(x_train)
    target = Var(y_train)

原因:定义数组的时候,数组是1*100的。转成tensor类型也是1*100的,但是由于线性回归,维度是1输入1输出的。因为应该设置成100*1来进行操作。

然后我又想展示以下数据,因此抄了一段程序https://blog.csdn.net/out_of_memory_error/article/details/81262309,把他改成了numpy,展示了一下。

x_train = torch.unsqueeze(torch.linspace(-1, 1, 100), dim=1)
y_train = 3 * x_train + 10 + torch.rand(x_train.size())
# 上面这行代码是制造出接近y=3x+10的数据集,后面加上torch.rand()函数制造噪音
# 上面是抄的 

#tensor 转numpy
x = x_train.numpy()
y = y_train.numpy()
plt.plot(x,y, 'ro', label='Original data')
plt.legend()
plt.show()

 

 

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Foneone

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值