pytorch框架学习——(2)

接下来以线性回归模型对pytorch框架进行一个介绍

在pytorch中对于任何一个模型都要先定义一个模型的内,在这个模型类中super下和前向传播中将会记录数据经过哪些层(卷积层,池化层等),这样就可以建立一个基本的模型

接下来指定参数和损失函数

首先是学习的次数和学习率,其次要指定优化器,pytorch中有很多种优化器可以选择,这里选择最基本的SGD就行,其次要指定损失函数,pytorch中也提供了非常多的损失函数,也可以自行选择。

然后定义一下参数

然后是训练参数,这里要注意每次把优化器清零和进行参数更新

然后预测一下(就是直接前向传播一下)

然后是保存和加载模型

可以发现模型是以字典的形式进行保存的

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值