epoch testing, model training stuck, will publish full version later,