pytorch

def validation_epoch_end(self, outputs):
    batch_losses = [x['val_loss']
    for x in outputs] epoch_loss = torch.stack(batch_losses).mean()
    # Combine losses batch_accs = [x['val_acc'] for x in outputs] epoch_acc = torch.stack(batch_accs).mean()
    # Combine accuracies return {'val_loss': epoch_loss.item(), 'val_acc': epoch_acc.item()}