              logger.trace("Training iteration: %s", iteration)
              save_iteration = iteration % self._args.save_interval == 0
              viewer = display_func if save_iteration or self._save_now else None
              timelapse = self._timelapse if save_iteration else None
              trainer.train_one_step(viewer, timelapse)
              if self._stop:
                  logger.debug("Stop received. Terminating")
                  break
              if save_iteration:
                  logger.trace("Save Iteration: (iteration: %s", iteration)
 
</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                 exit(1)
 </s> add                 sys.exit(1) </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)
            trainer.train_one_step(viewer, timelapse, self._args.colab)
              logger.trace("Training iteration: %s", iteration)
              save_iteration = iteration % self._args.save_interval == 0
              viewer = display_func if save_iteration or self._save_now else None
              timelapse = self._timelapse if save_iteration else None
            trainer.train_one_step(viewer, timelapse, self._args.colab)
              if self._stop:
                  logger.debug("Stop received. Terminating")
                  break
              if save_iteration:
                  logger.trace("Save Iteration: (iteration: %s", iteration)
keep keep keep keep replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
scripts/train.py