update

Scheaven

2021-09-18 291deeb1fcf45dbf39a24aa72a213ff3fd6b3405

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
B  
_àã@sLdZd
dl
Z
d
dlmZ
d
dlm
mZ
d
dlmZ
Gdd„dejƒZ    dS)z9
@author:  liaoxingyu
@contact: sherlockliao01@gmail.com
éN)
Ú    Parametercs,eZ
dZ‡f
d
d„Zdd„Zdd„Z‡ZS)ÚCirclecsDtƒ 
¡
||_||_|
jjj|_|
jjj|_    t
t ||¡ƒ
|_ dS)
N)ÚsuperÚ__init__Úin_featÚ_num_classesÚMODELÚHEADSÚSCALEÚ_sÚMARGINÚ_mrÚtorchÚTensorÚweight)ÚselfÚcfgrÚnum_classes)
Ú    __class__©ú./layers/circle.pyrs





zCircle.__init__cCs¤t 
t |
¡
t |j¡
¡}t | ¡d
|j¡
}t | ¡|j¡
}d
|j}|j}|j|||}|j|||}    tj||j    d}||d||    }
|
S)Né
)
rgð?)
Ú
FÚlinearÚ    normalizerÚreluÚdetachr rÚone_hotr)rÚfeaturesÚtargetsZsim_matZalpha_pZalpha_nZdelta_pZdelta_nZs_pZs_nÚpred_class_logitsrrrÚforwards






zCircle.forwardc

Csd
 |j
|j|j|j¡S)Nz3in_features={}, num_classes={}, scale={}, margin={})Úformatrrrr )
rrrrÚ
extra_repr's

zCircle.extra_repr)Ú__name__Ú
__module__Ú__qualname__rr!r#Ú __classcell__rr)
rrr s
    r)
Ú__doc__rÚtorch.nnÚnnÚtorch.nn.functionalÚ
functionalrrÚModulerrrrrÚ<module>s