It should be output_weights and not average.