model_summaries: fix weights in compute-summary

Fixed a bug when computing the weights of grouped convolutions (DWS).
This bug only appreared in the compute summary.
Replaced the computation with a simpler approach.