ankit01001

jax.core.UnexpectedTracerError while initializing weights into ReformerLM in Reformer: Text Generation colab

3

### Description a tracer error occurs inside jax while initializing pretrained weights in model.init_from_file() function inside this notebook https://colab.research.google.com/github/google/trax/blob/master/trax/models/reformer/text_generation.ipynb --------------------------------------------------------------------------- LayerError Traceback (most recent call last) in () 2 model...