GANExecutor: add constructor

zurutech · Sep 6, 2019 · f053532 · f053532
1 parent b22710e
commit f053532
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/ashpy/losses/gan.py b/ashpy/losses/gan.py
@@ -44,6 +44,18 @@ class GANExecutor(Executor, ABC):
     Implements the basic functions needed by the GAN losses.
     """
 
+    def __init__(self, fn: tf.keras.losses.Loss = None):
+        """Initialize the GANExecutor.
+
+        Args:
+            fn (:py:class:`tf.keras.losses.Loss`): A Keras Loss to execute.
+
+        Returns:
+            :py:obj:`None`
+
+        """
+        super().__init__(fn)
+
     @abstractmethod
     def call(self, context, **kwargs):
         """