nn.Module是所有神经网络层的父类。要实现自己的类的话,要先继承他。
特性:
1、有很多内置的函数可以直接调用:
2、Container 可以用nn.Sequential类
3、对网络内的参数更加高效的管理
4、对所有modules的结点式存储
sequential是第一个结点,其又有三个modules结点BasicNet、ReLU、Linear,BN又有一个线性层module
5 可以方便地将module类对象转移到不同的device上
6、save and load
可以在train的时候保存网络的中间状态
7、train/test的便捷切换
8、实现自己的layer