BaseModule是OpenMMLab中所有模块的基础模块,BaseModule包装了torch.nn.Module,添加了参数初始化函数等。
相比于torch.nn.Module,BaseModule主要加入了三个属性:
- init_cfg:控制初始化的config;
- init_weights:参数初始化函数,并记录初始化信息;
类在实例化时的参数有:
- init_cfg
可以查询的属性有:
- is_init
- init_cfg
可以调用的属性有:
- init_weights
- 直接print