I come upon the same issues using 0.4.1. What make me quite confused in that it will timeout once the second epoch of my model training begins, while the first epoch goes very smoothly. Do not what what is the reason…