def training_epoch_end(self, outputs):
    train_loss_mean = torch.stack([x['loss'] for x in outputs]).mean()
    self.logger.experiment.log({'train_loss_epoch': train_loss_mean}, step=self.examples_count)
def training_epoch_end(self, outputs):
    train_loss_mean = torch.stack([x['loss'] for x in outputs]).mean()
    self.logger.experiment.log({'train_loss_epoch': train_loss_mean}, step=self.examples_count)