def _setup_training(self) -> None:
    '\n        Setup the different steps to train a model:\n          - check if all the data are given\n          - create the proper strategy\n          - create the features\n          - prepare the model settings\n        '
    self._prepare_dataset()
    with self.args.strategy.scope():
        self._create_optimizer()
        _ = self.optimizer.iterations
        self._set_loss_and_metric()
        self._create_checkpoint_manager()
        self._create_summary_writer()
Setup the different steps to train a model:
  - check if all the data are given
  - create the proper strategy
  - create the features
  - prepare the model settings
def _setup_training(self) -> None:
    '\n        Setup the different steps to train a model:\n          - check if all the data are given\n          - create the proper strategy\n          - create the features\n          - prepare the model settings\n        '
    self._prepare_dataset()
    with self.args.strategy.scope():
        self._create_optimizer()
        _ = self.optimizer.iterations
        self._set_loss_and_metric()
        self._create_checkpoint_manager()
        self._create_summary_writer()