diff --git a/tensorflow_addons/callbacks/tqdm_progress_bar.py b/tensorflow_addons/callbacks/tqdm_progress_bar.py index 3d19e35524..78d1f66942 100644 --- a/tensorflow_addons/callbacks/tqdm_progress_bar.py +++ b/tensorflow_addons/callbacks/tqdm_progress_bar.py @@ -200,3 +200,17 @@ def format_metrics(self, logs={}, factor=1): metric_value_pairs.append(pair) metrics_string = self.metrics_separator.join(metric_value_pairs) return metrics_string + + def get_config(self): + config = { + 'metrics_separator': self.metrics_separator, + 'overall_bar_format': self.overall_bar_format, + 'epoch_bar_format': self.epoch_bar_format, + 'leave_epoch_progress': self.leave_epoch_progress, + 'leave_overall_progress': self.leave_overall_progress, + 'show_epoch_progress': self.show_epoch_progress, + 'show_overall_progress': self.show_overall_progress, + } + + base_config = super(TQDMProgressBar, self).get_config() + return dict(list(base_config.items()) + list(config.items()))