About 4 hours and 30 minutes of intense debugging. Finally managed to make it work. Still gotta make considerable architectural changes, and make sure it *converges* But hey! at least it is training and not throwing shape mismatches anymore.

→ View original post on X — @reach_vb