          logger.info("  Starting")
          if is_preview:
              logger.info("  Using live preview")
          logger.info("  Press '%s' to save and quit",
                      "Terminate" if self._args.redirect_gui else "ENTER")
          if not self._args.redirect_gui:
              logger.info("  Press 'S' to save model weights immediately")
          logger.info("===================================================")
  
          keypress = KBHit(is_gui=self._args.redirect_gui)
          err = False
 
</s> remove             exit(0)
 </s> add             sys.exit(0) </s> remove             trainer.train_one_step(viewer, timelapse)
 </s> add             trainer.train_one_step(viewer, timelapse, self._args.colab) </s> remove                 exit(1)
 </s> add                 sys.exit(1) </s> remove                 self.__print_loss(self._pingpong.loss)
 </s> add                 self.__print_loss(self._pingpong.loss, is_colab)
                    "Stop" if self._args.redirect_gui or self._args.colab else "ENTER")
        if not self._args.redirect_gui and not self._args.colab:
          logger.info("  Starting")
          if is_preview:
              logger.info("  Using live preview")
          logger.info("  Press '%s' to save and quit",
                    "Stop" if self._args.redirect_gui or self._args.colab else "ENTER")
        if not self._args.redirect_gui and not self._args.colab:
                    "Stop" if self._args.redirect_gui or self._args.colab else "ENTER")
        if not self._args.redirect_gui and not self._args.colab:
              logger.info("  Press 'S' to save model weights immediately")
          logger.info("===================================================")
  
          keypress = KBHit(is_gui=self._args.redirect_gui)
          err = False
keep keep keep keep replace replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
scripts/train.py