Where is the scheduled sampling constant beta in code? I don't see it. 