espnet2.uasr.loss.pseudo_label_loss.UASRPseudoLabelLoss

source

class espnet2.uasr.loss.pseudo_label_loss.UASRPseudoLabelLoss(weight: float = 1.0, input_dim: int = 128, output_dim: int = 64, downsample_rate: int = 2, ignore_index: int = -1, reduction: str = 'none')

Bases: AbsUASRLoss

auxiliary pseudo label loss for UASR.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(inter_x: Tensor, pseudo_labels: Tensor, is_discriminative_step: str2bool)

Forward.

Args: