Weights save, currently just using batch but only traiing a couple of things at once