You could write a weight_init function and call model.apply with it:
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)

net.apply(weights_init)