def update_running_loss(self):
    accumulated_loss = self.accumulated_loss.mean()
    if (accumulated_loss is not None):
        self.running_loss.append((self.accumulated_loss.mean() * self.trainer.accumulate_grad_batches))
    self.accumulated_loss.reset()
def update_running_loss(self):
    accumulated_loss = self.accumulated_loss.mean()
    if (accumulated_loss is not None):
        self.running_loss.append((self.accumulated_loss.mean() * self.trainer.accumulate_grad_batches))
    self.accumulated_loss.reset()