Source code for espnet2.spk.loss.abs_loss

#! /usr/bin/python
# -*- encoding: utf-8 -*-
# code from https://github.com/clovaai/voxceleb_trainer/blob/master/loss/aamsoftmax.py
# Adapted from https://github.com/wujiyang/Face_Pytorch (Apache License)
from abc import ABC, abstractmethod

import torch
import torch.nn as nn


[docs]class AbsLoss(nn.Module): def __init__(self, nout: int, **kwargs): super().__init__()
[docs] @abstractmethod def forward(self, x: torch.Tensor, label=None) -> torch.Tensor: raise NotImplementedError