17:全连接层创建

1:全连接网络的创建

17:全连接层创建

【注】

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

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

2:代码的实现

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

17:全连接层创建

 

 [注]nn.Sequential(

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

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

17:全连接层创建

 

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

两种风格的不同:

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

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

(2)步骤2:train

17:全连接层创建

 

17:全连接层创建

上一篇:静态内部类


下一篇:centos搭建本地yum源