def validation_step(self, batch, batch_idx, dataloader_idx):
    loss = self.model.get_loss(batch)
    result = self.model.result
    result.update({'val_loss': loss})
    return result
def validation_step(self, batch, batch_idx, dataloader_idx):
    loss = self.model.get_loss(batch)
    result = self.model.result
    result.update({'val_loss': loss})
    return result