Add an option to average the network parameters during training