Bugfix: Don't overide kernel_initializer if it is explicitly set in m…

…odel definition
deepfakes · Sep 27, 2019 · 84932c0 · 84932c0
1 parent 46efae3
commit 84932c0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/model/nn_blocks.py b/lib/model/nn_blocks.py
@@ -53,6 +53,9 @@ def set_default_initializer(self, kwargs):
             to conv_aware or he_uniform().
             if a specific initializer has been passed in then the specified initializer
             will be used rather than the default """
+        if "kernel_initializer" in kwargs:
+            logger.debug("Using model specified initializer: %s", kwargs["kernel_initializer"])
+            return kwargs
         if self.use_convaware_init:
             default = ConvolutionAware()
             if self.first_run: