upd comment

2019-04-24 11:41:10 +04:00
parent b6711b97a3
commit 0cd8dd7296
1 changed files with 1 additions and 0 deletions
@@ -591,6 +591,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
        nnlib.Adam = Adam

        def CAInitializerMP( conv_weights_list ):
+            #Convolution Aware Initialization https://arxiv.org/abs/1702.06295            
            result = CAInitializerMPSubprocessor ( [ (i, K.int_shape(conv_weights)) for i, conv_weights in enumerate(conv_weights_list) ], K.floatx(), K.image_data_format() ).run()
            for idx, weights in result:
                K.set_value ( conv_weights_list[idx], weights )