Hi @Hamid
I found that weight initialization can by done as below in the latest version of pytorch
import torch.nn.init as init
self.conv1 = nn.Conv2d(3, 20, 5, stride=1, bias=True)
init.xavier_uniform(self.conv1.weight, gain=np.sqrt(2.0))
init.constant(self.conv1.bias, 0.1)