I'm retraining and the graph looks much better