Method __init__

cdslib/core/nn/modules/vrnn.py:210–232

Create a Variational Recurrent Neural Network (VRNN) model. Args: param_dict: A :py:class:`ParamVRNN` object to define the hyper-parameters of the network. kwargs: If param_dict is None, you can directly provide keyword argume

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

    """

    def __init__(self, param_dict: ParamVRNN = None, **kwargs):
        """
        Create a Variational Recurrent Neural Network (VRNN) model.

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

        # read and set configs
        if param_dict is not None:
            self.config_dict = ParamVRNN(**param_dict)
        else:
            self.config_dict = ParamVRNN(**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.