o ϯiO@s8ddlmZdejddfddZdejddfddZdS)NmodulereturncCs4tjj|jdd|jdurtj|jddSdS)z Initialize `module.weight` using the "XavierFill" implemented in Caffe2. Also initializes `module.bias` to 0. Args: module (torch.nn.Module): module to initialize. )aNr)nninitkaiming_uniform_weightbias constant_rr I/home/ubuntu/.local/lib/python3.10/site-packages/fvcore/nn/weight_init.pyc2_xavier_fills rcCs6tjj|jddd|jdurtj|jddSdS)z Initialize `module.weight` using the "MSRAFill" implemented in Caffe2. Also initializes `module.bias` to 0. Args: module (torch.nn.Module): module to initialize. fan_outrelu)mode nonlinearityNr)rrkaiming_normal_r r r r r r r c2_msra_fills r)torch.nnrModulerrr r r rs