code: def set_generator_training_nb(self, number):
          
  
          self.samples_per_epoch = number
          diff_to_batch = number % self.get_batch_size()
          if diff_to_batch > 0:
              self.samples_per_epoch += self.get_batch_size() - diff_to_batch
docs: sets self.samples_per_epoch which is used in model.fit if input is a generator
          :param number:
          :return: