clarity.engine.losses module

class clarity.engine.losses.SISNRLoss(*args: Any, **kwargs: Any)

Bases: Module

cal_sisnr(x, s, eps=1e-08)

Arguments: x: separated signal, N x S tensor s: reference signal, N x S tensor Return: sisnr: N tensor

forward(x, y)

class clarity.engine.losses.SNRLoss(*args: Any, **kwargs: Any)

Bases: Module

forward(x, s, eps=1e-08)

l2norm(mat, keepdim=False)

class clarity.engine.losses.STOILevelLoss(*args: Any, **kwargs: Any)

Bases: Module

alpha

rms measurement

forward(x, s)

gamma_a

mse

measure_loudness(signal, eps=1e-08)

class clarity.engine.losses.STOILoss(*args: Any, **kwargs: Any)

Bases: Module

forward(x, s)