Sth Like that?
````` def weights_init(m):
          with torch.no_grad():
                if isinstance(m, nn.Conv3d):
                       torch.nn.init.normal_(m.weight)
                       torch.nn.init.normal_(m.bias)
model = ConvNetRedo1(32,64,(7,7,5),(5,5,3),500,100)
model.apply(weight_init)
model=model.cuda
for epoch in range(num_epochs):
       for i, data  in enumerate(trainloader,0):
              images, labels=data   
               optimizer.zero_grad()
                outputs= model(images)````