Try with only 1 epoch