Пример #1
0
 def test_kaiming_uniform_initializer_static(self):
     """Test Kaiming unorm initializer for matrix multiply.
     """
     self.static_test_kaiming_initializer_common(
         init_inst=initializer.KaimingUniform(),
         dtype="float32",
         uniform=True,
         is_conv=False)
Пример #2
0
 def test_kaiming_uniform_initializer_conv_static(self):
     """Test Kaiming unorm initializer for convolutions.
     """
     self.static_test_kaiming_initializer_common(
         init_inst=initializer.KaimingUniform(),
         dtype="float32",
         uniform=True,
         is_conv=True)
Пример #3
0
 def test_kaiming_dygraph(self):
     self.dygraph_test_kaiming_initializer_common(
         init_inst=initializer.KaimingUniform(),
         dtype="float32",
         uniform=True)
     self.dygraph_test_kaiming_initializer_common(
         init_inst=initializer.KaimingNormal(),
         dtype="float32",
         uniform=False)