pytorch学习

目前的状态:人有点晕。好多细节的东西都不知道怎么来的。大方向有所把握:

1. 准备数据集:看起来很简单,其实不然。如何把文件读取进来,变成pytorch所需要的数据类型。

图片:你就需要ToTensor,Normalize转换为需要的数据类型

文字:对init,getitem,len进行重写

准备dataset,构建data_loader并返回

2. 构建模型:重写init和forward方法。在forward里对每一层进行处理。包括矩阵变换,激活函数等去得到输出

3. 训练:基本就是循环里面梯度归零,调用,loss,反向传播,更新

data_loader = get_dataloader()
    for idx,(input,traget) in enumerate(data_loader):
        optimizer.zero_grad()   # 梯度归零
        output = model(input)    # 调用模型得到预测值
        loss = F.nll_loss(output,traget)   # 得到损失
        loss.backward()   # 反向传播
        optimizer.step()   # 梯度更新

4. 测试:pass

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值