add prepare epoch
write prepare.py