美文网首页
Pytorch袖珍手册之九

Pytorch袖珍手册之九

作者: 深思海数_willschang | 来源:发表于2021-08-25 14:24 被阅读0次
    pytorch pocket reference

    第五章 基于Pytorch的深度学习网络结构自主式开发

    前面章节我们主要通过pytorch提供的类,函数和各种库进行模型设计及训练,在本章节中我们将通过几个代码片断来向大家展示pytorch的灵活之处,即用户可以基于pytorch的一些基础函数构建自己的新模型,激活函数,优化器等模块。

    how to create your own custom deep learning components and algorithms in PyTorch.

    • 自定义神经网络层和激活函数

    If we take a look at the PyTorch source code, we’ll see that layers and activations are created using a functional definition anda class implementation.

    书中通过对线性连接层(torch.nn.function.linear / torch.nn.Linear)的源码进行解读,说明pytorch中各layers是怎么构建的。同时自定义了一个复数形式的线性层。

    自定义激活函数就比较简单,即根据激活函数算法进行编程实现。

    • 自定义模型结构
      根据实际应用需要,可以基于nn.Module自定义搭建不同结构的模型框架,主要就是完善init()和forward()函数。

    • 自定义损失函数

    • 自定义优化算法

    • 自定义训练,验证及测试循环机制

    这章节内容比较简单,到时回看书里原文即可。

    原书下载地址:
    我用阿里云盘分享了「OReilly.PyTorch.Pocket.R...odels.149209000X.pdf」,你可以不限速下载🚀
    复制这段内容打开「阿里云盘」App 即可获取
    链接:https://www.aliyundrive.com/s/NZvnGbTYr6C

    相关文章

      网友评论

          本文标题:Pytorch袖珍手册之九

          本文链接:https://www.haomeiwen.com/subject/pkitiltx.html