fixed the pre-training epoches