对于python/pytorch初学者,可以先熟悉python中类的使用,以及nn.Module模块的相关定义
- python中类的init()的两种定义方式:python def init(self, name等多参数), def init(self)
- super()以及python2和python3中使用super()的不同点:python中super().init()
- 在继承 nn.Module 的神经网络模块在实现自己的 init 函数时,一定要先调用 super().init()。只有这样才能正确地初始化自定义的神经网络模块,否则会缺少相关成员变量而导致模块被调用时出错。PyTorch 源码解读之 nn.Module:核心网络模块接口详解