pytorch使用——（六）模型创建与nn.Module

1、网络模型创建步骤

2、代码逻辑 

3、torch.nn简介

4、nn.Module

一个module可以包含多个子module；一个module相当于一个运算，必须实现forward()函数；每个module都有8个字典管理它的属性。具体如下：

self._parameters = OrderedDict()
self._buffers = OrderedDict()
self._backward_hooks = OrderedDict()
self._forward_hooks = OrderedDict()
self._forward_pre_hooks = OrderedDict()
self._state_dict_hooks = OrderedDict()
self._load_state_dict_pre_hooks = OrderedDict()
self._modules = OrderedDict()
           

• parameters: 存储管理nn.Parameter类
• modules : 存储管理nn.Module类
• buffers：存储管理缓冲属性，如BN层中的running_mean
• ***_hooks：存储管理钩子函数