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)