trax
trax copied to clipboard
Transformer classifier used to work with 1.3.3 now throws shape error after one iteration!
Description
I'm running the following Collab for transformer classifier. It used to work on version 1.3.3, but now throws shape error after one iteration!
https://colab.research.google.com/drive/1sNsYGeBsPQJLhSGKBur8C77jpMWH90Bm?usp=sharing