After the model is created, initialize the variables in con2d, convtranspose2d, and normalized batchnorm2d functions

 def weights_init(m):
     classname = m.__class__.__name__
     if (classname.find('Conv2d') != -1) | (classname.find('ConvTranspose2d') != -1):
         m.weight.data.normal_(0.0, 0.02)
     elif classname.find('BatchNorm2d') != -1:
         m.weight.data.normal_(1.0, 0.02)
         m.bias.data.fill_(0)

#  Defining models 
m = model()
m.cuda()
#  Model initialization 
weights_init(m)