Method __init__

cdslib/core/nn/modules/graves.py:198–219

Create an Alex Graves' model. Args: param_dict: A :py:class:`ParamGraves` object to define the hyper-parameters of the network. kwargs: If param_dict is None, you can directly provide keyword arguments of :py:class:`ParamGraves` here.

(self, param_dict: ParamGraves = None, **kwargs)

    """

    def __init__(self, param_dict: ParamGraves = None, **kwargs):
        """Create an Alex Graves' model.

        Args:
            param_dict:
                A :py:class:`ParamGraves` object to define the hyper-parameters of the network.
            kwargs:
                If param_dict is None, you can directly provide keyword arguments of :py:class:`ParamGraves` here.
        """
        super().__init__()

        # read and set configs
        if param_dict is not None:
            self.config_dict = ParamGraves(**param_dict)
        else:
            self.config_dict = ParamGraves(**kwargs)

        for key in self.config_dict:
            setattr(self, key, self.config_dict[key])

        # construct sub-networks
        self._construct_networks()

    def _construct_networks(self):
        """Construct the sub-networks.