[Pytorch] How to Apply the Weight Initialization (Code)

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

docs.ray.io/en/master/tune/tutorials/tune-advanced-tutorial.html

Guide to Population Based Training (PBT) — Ray v1.2.0.dev0