def get_config(self): config = { "return_selection": self.return_selection, } for key in self.kwargs_keys: config[key] = serialize_kwarg(key, getattr(self, key)) base_config = super().get_config() return {**base_config, **config}
def get_config(self): mp_config = {"aggregate": serialize_scatter(self.agg)} keras_config = {} for key in self.kwargs_keys: keras_config[key] = serialize_kwarg(key, getattr(self, key)) base_config = super().get_config() return {**base_config, **keras_config, **mp_config, **self.config}
def get_config(self): config = { 'aggregate': self.agg, } for key in self.kwargs_keys: config[key] = serialize_kwarg(key, getattr(self, key)) base_config = super().get_config() return {**base_config, **config}
def get_config(self): base_config = super().get_config() keras_config = {} for key in self.kwargs_keys: keras_config[key] = serialize_kwarg(key, getattr(self, key)) return {**base_config, **keras_config, **self.config}