                  model.save_models()
                  trainer.clear_tensorboard()
              except KeyboardInterrupt:
                  logger.info("Saving model weights has been cancelled!")
              exit(0)
          except Exception as err:
              raise err
  
      def _load_model(self):
          """ Load the model requested for training.
 
</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     def __print_loss(self, loss):
 </s> add     def __print_loss(self, loss, is_colab): </s> remove     def train_one_step(self, viewer, timelapse_kwargs):
 </s> add     def train_one_step(self, viewer, timelapse_kwargs, is_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)
            sys.exit(0)
                  model.save_models()
                  trainer.clear_tensorboard()
              except KeyboardInterrupt:
                  logger.info("Saving model weights has been cancelled!")
            sys.exit(0)
          except Exception as err:
              raise err
  
      def _load_model(self):
          """ Load the model requested for training.
keep keep keep keep replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
scripts/train.py