losses.py

import torch


def l1(output, target):
    return torch.mean(torch.abs(output - target))


def l1_wav(output_dict, target_dict):
	return l1(output_dict['segment'], target_dict['segment'])


def get_loss_function(loss_type):
    if loss_type == "l1_wav":
        return l1_wav

    else:
        raise NotImplementedError("Error!")