Anyone know a TI training repo that's a bit more memory optimized than the original one?

It's a bit kludgy getting it working on a 12 GB gpu, I managed it but I'd rather a less hacky solution.