Hi Yuhuai,

In ais.py you commented under the sigmoid_schedule() function that

The sigmoid schedule defined in Section 6.2 of the paper

but there is no 6.2 in the paper tho.

It will be great if you can provide some pointer, I learned a lot from your code! A great contribution to the community.

Thanks!
Xuan