longer training + predict.py fix