AudioSep / losses.py
badayvedat's picture
Initial commit
ae29df4
raw
history blame contribute delete
331 Bytes
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!")