config = {
        "vocab_size":
            self._vocab_size,
        "hidden_size":
            self._hidden_size,