So we would have two options, one which would give the mean of output_weights = None, and the other as they are without averaging. @arjoly , Your thoughts also please?