          for side in ("a", "b"):
              image_dir = getattr(self._args, "input_{}".format(side))
              if not os.path.isdir(image_dir):
                  logger.error("Error: '%s' does not exist", image_dir)
                  exit(1)
  
              images[side] = get_image_paths(image_dir)
              if not images[side]:
                  logger.error("Error: '%s' contains no images", image_dir)
                  exit(1)
 
</s> remove                 exit(1)
 </s> add                 sys.exit(1) </s> remove                     "Terminate" if self._args.redirect_gui else "ENTER")
        if not self._args.redirect_gui:
 </s> add                     "Stop" if self._args.redirect_gui or self._args.colab else "ENTER")
        if not self._args.redirect_gui and not self._args.colab: </s> remove         print("[{}] [#{:05d}] {}".format(self._timestamp,
                                         self._model.iterations,
                                         output), end='\r')
 </s> add         output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output) </s> remove                 self.__print_loss(loss)
 </s> add                 self.__print_loss(loss, is_colab) </s> remove                 self.__print_loss(self._pingpong.loss)
 </s> add                 self.__print_loss(self._pingpong.loss, is_colab)
                sys.exit(1)
          for side in ("a", "b"):
              image_dir = getattr(self._args, "input_{}".format(side))
              if not os.path.isdir(image_dir):
                  logger.error("Error: '%s' does not exist", image_dir)
                sys.exit(1)
  
              images[side] = get_image_paths(image_dir)
              if not images[side]:
                  logger.error("Error: '%s' contains no images", image_dir)
                  exit(1)
keep keep keep keep replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
scripts/train.py