As I was having the same issues as you and that’s the reason why I stopped training new models using legacy TT2