              script = self.import_script()
              process = script(arguments)
              process.process()
          except FaceswapError as err:
              logger.error(str(err))
              crash_file = crash_log()
              logger.info("To get more information on this error see the crash report written to "
                          "%s", crash_file)
          except KeyboardInterrupt:  # pylint: disable=try-except-raise
              raise
 
</s> remove         loss = self.model.predictors[self.side].train_on_batch(*batch)
 </s> add         try:
            loss = self.model.predictors[self.side].train_on_batch(*batch)
        except tf_errors.ResourceExhaustedError as err:
            msg = ("You do not have enough GPU memory available to train the selected model at "
                   "the selected settings. You can try a number of things:"
                   "\n1) Close any other application that is using your GPU (web browsers are "
                   "particularly bad for this)."
                   "\n2) Lower the batchsize (the amount of images fed into the model each "
                   "iteration)."
                   "\n3) Try Memory Saving Gradients and/or Ping Pong Training."
                   "\n4) Use a more lightweight model, or select the model's 'LowMem' option "
                   "(in config) if it has one.")
            raise FaceswapError(msg) from err </s> remove from lib.utils import get_folder, get_image_paths
 </s> add from lib.utils import FaceswapError, get_folder, get_image_paths
            for line in str(err).splitlines():
                logger.error(line)
              script = self.import_script()
              process = script(arguments)
              process.process()
          except FaceswapError as err:
            for line in str(err).splitlines():
                logger.error(line)
              crash_file = crash_log()
              logger.info("To get more information on this error see the crash report written to "
                          "%s", crash_file)
          except KeyboardInterrupt:  # pylint: disable=try-except-raise
              raise
keep keep keep keep replace keep keep keep keep keep
 Useful logging message for OOM 
lib/cli.py