Hi, I just found this bug, so I haven’t been able to reproduce it. I’ll try training without changing anything and see what happens. Only problem is that it takes around 6 hours to train a single epoch, so it’ll probably take some time.