fetch weight distribution for baselines as well