save image of weights