add 3d, height difference if needed, Tested with AutismMix
All the details in my article
Test my models in tensor