Confused regarding Model in Assignment UNQ 4

Course Q&A Natural Language Processing NLP with Attention Models
1

I have create the model using:
trax.models.reformer.ReformerLM() class, however the model output deviates from the expected output.

Screenshot from 2023-11-21 10-52-55
Screenshot from 2023-11-21 10-52-551916×963 91.2 KB

screencapture-coursera-org-learn-attention-models-in-nlp-programming-v6V4M-chatbot-lab-2023-11-21-10_44_41
screencapture-coursera-org-learn-attention-models-in-nlp-programming-v6V4M-chatbot-lab-2023-11-21-10_44_411642×7657 347 KB

2

Resolved this was caused as we need explicit variable declaration when using:
trax.models.reformer.ReformerLM()
like vocab_size=vocab_size etc. rather than passing vocab_size directly as a param