Is there a way to save the parameters of a Bayesian neural network (each neuron has two parameters, mean and variance) as we save the parameters of  a normal neural network by state_dict() ? is there any easier way than overwriting existing classes and modules of Pytorch?