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)
def __init__(self, loss_fn_config): self.config = loss_fn_config NLLLoss.__init__(self)