how many epochs to train mnist