tensorflow: the loss goes up at the beginning of each epoch