
        # load weights for transfer learning
        if self.transfer_learning:
            model.load_weights(load_weight_file_path(self.transfer_model_name))

        # for outer_loop in tqdm(range(outer_training_steps)):
        for steps in range(self.training_steps):
            # fetch training data