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
原文:https://www.cnblogs.com/jiafeng1996/p/15087481.html