code: def run(self):
          
          _, test_data = self.data.load(train=False, test=True)
          try:
              self.model.fit_generator(
                  self.samples_to_batches(self.generate_samples(), self.args.batch_size), steps_per_epoch=self.args.steps_per_epoch,
                  epochs=self.epoch + self.args.epochs, validation_data=test_data,
                  callbacks=self.callbacks, initial_epoch=self.epoch
              )
          finally:
              self.model.save(self.args.model)
              save_params(self.args.model)
docs: Train the model on randomly generated batches