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}
