o
    pih                     @   s(   d dl Z ddlmZ G dd deZdS )    N   )LargeMarginSoftmaxLossc                   @   s   e Zd Zdd ZdS )SphereFaceLossc                 C   s$   t j|ddd}||d | j S )N   r   )pdim)torchnorm	unsqueezescale)selflogits
embeddingsembedding_norms r   b/home/ubuntu/.local/lib/python3.10/site-packages/pytorch_metric_learning/losses/sphereface_loss.pyscale_logits   s   zSphereFaceLoss.scale_logitsN)__name__
__module____qualname__r   r   r   r   r   r      s    r   )r   large_margin_softmax_lossr   r   r   r   r   r   <module>   s    