          logger.trace(loss)
          output = ["Loss {}: {:.5f}".format(side.capitalize(), loss[side][0])
                    for side in sorted(loss.keys())]
          output = ", ".join(output)
          print("[{}] [#{:05d}] {}".format(self._timestamp,
                                           self._model.iterations,
                                           output), end='\r')
  
      def train_one_step(self, viewer, timelapse_kwargs):
          """ Running training on a batch of images for each side.
  
          Triggered from the training cycle in :class:`scripts.train.Train`.
 
</s> remove     def train_one_step(self, viewer, timelapse_kwargs):
 </s> add     def train_one_step(self, viewer, timelapse_kwargs, is_colab): </s> remove                 exit(1)
 </s> add                 sys.exit(1) </s> remove     def __print_loss(self, loss):
 </s> add     def __print_loss(self, loss, is_colab):
        output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output)
          logger.trace(loss)
          output = ["Loss {}: {:.5f}".format(side.capitalize(), loss[side][0])
                    for side in sorted(loss.keys())]
          output = ", ".join(output)
        output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output)
        output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output)
        output = "[{}] [#{:05d}] {}".format(self._timestamp, self._model.iterations, output)
        if not is_colab:
            print(output, end='\r')
        else:
            print(output)
  
      def train_one_step(self, viewer, timelapse_kwargs):
          """ Running training on a batch of images for each side.
  
          Triggered from the training cycle in :class:`scripts.train.Train`.
keep keep keep keep replace replace replace keep keep keep keep keep
 lib,cli: Add suppressed colab flag 
plugins/train/trainer/_base.py