首页 > 其他 > 详细

17:全连接层创建

时间:2021-08-01 22:34:35      阅读:22      评论:0      收藏:0      [点我收藏+]

1:全连接网络的创建

技术分享图片

【注】

(1):自己创建的网络结构需要继承nn.Module

(2):需要实现forward()函数,不需要实现backward(),因为nn.Module提供了backward()函数的实现。(pytorch的autograd包会自动完成向后求导的过程。)

2:代码的实现

(2.1)步骤一:全连接网络的创建以及前向传播的实现

技术分享图片

 

 [注]nn.Sequential(

)类似于一个容器,容器中可以添加任何继承自nn.Module的类。也可以添加自己创建的类。

[注]self.model继承自nn.Module故可以使用self.model(x)调用model.forward()函数。

技术分享图片

 

 [注]:nn.ReLU为类风格的API,F.relu()为函数风格的API。

两种风格的不同:

对于类风格的API必须先进行实例化,再进行调用。并且其内部参数w,b必须通过para方法来进行访问。

对于函数风格的API可以自己进行过程的管理,仅仅使用了gpu加速的功能。

(2)步骤2:train

技术分享图片

 

17:全连接层创建

原文:https://www.cnblogs.com/jiafeng1996/p/15087481.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!