remove limit on training data - train on all of bach