Can you move the description of ``output_weights`` to the appropriate parameter?