Increased default epochs to 2000 instead of 500, I think that fixed it but let me know if you still run into this.