SubhadityaMukherjee/weights.md

Created February 13, 2020 06:43

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/SubhadityaMukherjee/496e9767b55644f40419ec19ecb6dcd1.js"></script>
Save SubhadityaMukherjee/496e9767b55644f40419ec19ecb6dcd1 to your computer and use it in GitHub Desktop.

Raw

def weights_init(m):
    classname = m.__class__.__name__
    if classname.find('Conv') !=-1:
        nn.init.normal_(m.weight.data,0.0,.02)
    elif classname.find('BatchNorm') !=-1:
        nn.init.normal_(m.weight.data,1.0,.02)
        nn.init.constant_(m.bias.data,0)