UNQ_C3, 5 tests passed, 1 test failed

While running the training loop for NER, I pass 5 tests out of 6.
In order to properly understand the issue to I coded everything on colab. Here is the error I am getting.

Can anyone help me find the bug…

Hi @Rvi

The problem might lie in data_generator implementation. Do the outputs match exactly the expected outputs?