adding some plots of change in weights over training time