mmcv中BaseModule和Pytorch的nn.Module的区别

1、mmcv的BaseModule

BaseModule类在mmcv.runner中

比起torch.nn.Module类而言,多了init_cfg和init_weights

所有的init方法都是一个INITIALIZERS

注册在mmcv/cnn/utils/weight_init.py

基础知识:

normal是正态分布

uniform是均匀分布

constant是初始化为常数

type里的override是用来init这个实例里一些特殊的部分,比如self.reg

2、Pytorch的nn.Module

Module类在torch.nn中

这个实例中的可学习参数的初始化方式为

norm的weight都是1,bias都是0

猜你喜欢

转载自blog.csdn.net/jiangqixing0728/article/details/130184488
今日推荐