Exemple #1
0
    def __init__(self, name: Optional[str] = None):
        """
        Constructor.

        Args:
            name: Name of the module (DEFAULT: None)
        """
        # Call the base constructors.
        # Serialization.__init__(self, name=name)
        torch_NLLLoss.__init__(self)
Exemple #2
0
 def __init__(self, loss_fn_config):
     self.config = loss_fn_config
     NLLLoss.__init__(self)