Ejemplos de caffe2_xavier_init en Python

Ejemplo n.º 1

Archivo: hrfpn.py Proyecto: ZDstandup/LIO

 def init_weights(self):
     for m in self.modules():
         if isinstance(m, nn.Conv2d):
             caffe2_xavier_init(m)

Ejemplo n.º 2

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