mention weights in print method for model summaries