def init_weights(self): for m in self.modules(): if isinstance(m, nn.Conv2d): caffe2_xavier_init(m)
def init_weights(self): print("init weights in dlafpn") for m in self.modules(): if isinstance(m, nn.Conv2d): caffe2_xavier_init(m)