Python caffe2_xavier_init примеры использования

Пример #1

Файл: hrfpn.py Проект: ZDstandup/LIO

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

Пример #2

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