trax
trax copied to clipboard
Reformer: Machine Translation can not run in colab
Description
Wen I run this machine_translation.ipynb,it not work
Error logs:
TypeError: reshape total size must be unchanged, got new_sizes (1, 1, 1, 128) for shape (1, 1024, 128).