a  þd ã@s<ddlZddlmZGdd„dejƒZGdd„dejƒZdS)éNcs&eZdZd ‡fdd„ Zdd „Z‡ZS) ÚAsymmetricLossMultiLabelééçš™™™™™©?ç:Œ0âŽyE>Fcs0tt|ƒ ¡||_||_||_||_||_dS)N)ÚsuperrÚ__init__Ú gamma_negÚ gamma_posÚclipÚdisable_torch_grad_focal_lossÚeps)Úselfr r r r r ©Ú __class__©úb/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/loss/asymmetric_loss.pyrs z!AsymmetricLossMultiLabel.__init__cCst |¡}|}d|}|jdur<|jdkr<||jjdd}|t |j|jd¡}d|t |j|jd¡}||}|jdksŒ|jdkrü|jržtj   d¡||} |d|} | | } |j||jd|} t  d| | ¡} |jrôtj   d¡|| 9}|  ¡ S)zz" Parameters ---------- x: input logits y: targets (multi-label binarized vector) rNr)Úmax)ÚminFT) ÚtorchZsigmoidr ÚclampÚlogr r r r Z_CZset_grad_enabledÚpowÚsum)rÚxÚyZ x_sigmoidÚxs_posÚxs_negZlos_posZlos_negÚlossZpt0Zpt1ÚptZone_sided_gammaZ one_sided_wrrrÚforwards(    z AsymmetricLossMultiLabel.forward)rrrrF)Ú__name__Ú __module__Ú __qualname__rr Ú __classcell__rrrrrs rcs.eZdZd edœ‡fdd„ Zd d d „Z‡ZS) ÚAsymmetricLossSingleLabelrrçš™™™™™¹?Úmean)r cs>tt|ƒ ¡||_tjdd|_g|_||_||_ ||_ dS)Néÿÿÿÿ©Zdim) rr%rr ÚnnZ LogSoftmaxÚ logsoftmaxÚtargets_classesr r Ú reduction)rr r r r-rrrr6sz"AsymmetricLossSingleLabel.__init__Nc Csì| ¡d}| |¡}t |¡ d| ¡ d¡d¡|_|j}d|}t |¡}d|} ||}| |} t  d|| |j ||j |¡} || }|j dkr¼|j  d|j ¡ |j |¡|_|j  |¡ } | jdd} |jdkrè|  ¡} | S)zj" Parameters ---------- x: input logits y: targets (1-hot vector) r(rrr)r')Úsizer+rZ zeros_likeZscatter_ÚlongZ unsqueezer,Úexprr r r ÚmulÚaddrr-r') rÚinputsÚtargetr-Z num_classesZ log_predsÚtargetsZ anti_targetsrrZ asymmetric_wrrrrr @s(    ÿ    z!AsymmetricLossSingleLabel.forward)rrr&r')N)r!r"r#Úfloatrr r$rrrrr%5s r%)rZtorch.nnr*ÚModulerr%rrrrÚs 0