NaN values in the results of validation and testing

[aTongs1]

Dear author,
Thank you very much for your work. I downloaded your code and started debugging it. I noticed that there are nan values in the results of validation and testing in notebooks/mamba.ipynb. Could you please explain the reasons behind this?

[Wangzi06]

Did you ever fix this issue