B
|
©r_ ã @ s4 d Z ddlZddlmZ ddgZdd Zdd ZdS )z:
|
@author: xingyu liao
|
@contact: sherlockliao01@gmail.com
|
é N)ÚnnÚweights_init_classifierÚweights_init_kaimingc C s¼ | j j}| d¡dkrDtj | jdd¡ | jd k r¸tj | jd¡ nt| d¡dkrtjj | jddd | jd k r¸tj | jd¡ n6| d
|
¡dkr¸| j
|
r¸tj | jd d ¡ tj | jd¡ d S ) NÚLinearéÿÿÿÿr g{®Gáz?g ZConvÚfan_outÚrelu)ÚmodeÚ nonlinearityÚ BatchNormg ð?g{®Gáz?) Ú __class__Ú__name__Úfindr ÚinitÚnormal_ÚweightÚbiasÚ constant_Úkaiming_normal_Úaffine)ÚmÚ classname© r ú./utils/weight_init.pyr s
|
|
c C sF | j j}| d¡dkrBtjj| jdd | jd k rBtj | jd¡ d S )Nr r gü©ñÒMbP?)Ústdg ) r r r r r r r r r )r r r r r r s
|
|
)Ú__doc__ÚmathÚtorchr Ú__all__r r r r r r Ú<module> s
|
