def METHOD_NAME(self, batch, batch_idx):
    x, y = batch
    y_hat = self(x)
    loss = torch.nn.MSELoss()(y_hat, y)
    self.log('phase', 'train')
    self.log('loss', loss)
    self.log('count', len(y))
    self.log('time', str(datetime.datetime.now()))
    return loss